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

fix wrong parameter

fix a bug on `torch_dtype` argument in `from_single_file` of ControlNetModel (#5528)
fix wrong parameter
dbce14da · Yi-Xuan XU · GitHub · 71ad0260 · dbce14da
Unverified Commit dbce14da authored Oct 25, 2023 by Yi-Xuan XU Committed by GitHub Oct 25, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/loaders.py src/diffusers/loaders.py +1 -1

No files found.
--- a/src/diffusers/loaders.py
+++ b/src/diffusers/loaders.py
@@ -3171,7 +3171,7 @@ class FromOriginalControlnetMixin:
        )

        if torch_dtype is not None:
-            controlnet.to(torch_dtype=torch_dtype)
+            controlnet.to(dtype=torch_dtype)

        return controlnet