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