From dbce14da5635cd511eae19bc535104e188642082 Mon Sep 17 00:00:00 2001 From: Yi-Xuan XU Date: Wed, 25 Oct 2023 23:29:56 +0800 Subject: [PATCH] fix a bug on `torch_dtype` argument in `from_single_file` of ControlNetModel (#5528) fix wrong parameter --- src/diffusers/loaders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/loaders.py b/src/diffusers/loaders.py index 2987fb718e..67043866be 100644 --- 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