Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "fee6d12c580b862d7818cbb0ccae6df9fc5e7064"
Unverified Commit dbce14da authored by Yi-Xuan XU's avatar Yi-Xuan XU Committed by GitHub
Browse files

fix a bug on `torch_dtype` argument in `from_single_file` of ControlNetModel (#5528) 

fix wrong parameter
parent 71ad0260
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/loaders.py
View file @ dbce14da
...@@ -3171,7 +3171,7 @@ class FromOriginalControlnetMixin: ...@@ -3171,7 +3171,7 @@ class FromOriginalControlnetMixin:
) )
if torch_dtype is not None: if torch_dtype is not None:
controlnet.to(torch_dtype=torch_dtype) controlnet.to(dtype=torch_dtype)
return controlnet return controlnet
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment