fix a bug on torch_dtype argument in from_single_file of ControlN…

…etModel (huggingface#5528) fix wrong parameter
yoonseokjin · Oct 25, 2023 · 37840e8 · 37840e8
1 parent 49615bf
commit 37840e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/loaders.py b/loaders.py
@@ -3171,7 +3171,7 @@ def from_single_file(cls, pretrained_model_link_or_path, **kwargs):
         )
 
         if torch_dtype is not None:
-            controlnet.to(torch_dtype=torch_dtype)
+            controlnet.to(dtype=torch_dtype)
 
         return controlnet