Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Style quickfix

This commit is contained in:
anton-l 2022-08-22 18:40:04 +02:00
parent db5fa43079
commit eb5267f377
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/modeling_utils.py
View File

@ -337,7 +337,9 @@ class ModelMixin(torch.nn.Module):
)
if torch_dtype is not None and not isinstance(torch_dtype, torch.dtype):
raise ValueError(f"{torch_dtype} needs to be of type `torch.dtype`, e.g. `torch.float16`, but is {type(torch_dtype)}.")
raise ValueError(
f"{torch_dtype} needs to be of type `torch.dtype`, e.g. `torch.float16`, but is {type(torch_dtype)}."
)
elif torch_dtype is not None:
model = model.to(torch_dtype)