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

xuyxu · web-flow · commit dbce14da5635 · 2023-10-25T17:29:56.000+02:00
fix wrong parameter
diff --git a/src/diffusers/loaders.py b/src/diffusers/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
 

Original file line number	Diff line number	Diff line change
`@@ -3171,7 +3171,7 @@ def from_single_file(cls, pretrained_model_link_or_path, **kwargs):`
`3171`	`3171`	`)`
`3172`	`3172`
`3173`	`3173`	`if torch_dtype is not None:`
`3174`		`- controlnet.to(torch_dtype=torch_dtype)`
	`3174`	`+ controlnet.to(dtype=torch_dtype)`
`3175`	`3175`
`3176`	`3176`	`return controlnet`
`3177`	`3177`