diff --git a/src/diffusers/loaders/single_file.py b/src/diffusers/loaders/single_file.py index 8c63c4cf59..bf100e7f2c 100644 --- a/src/diffusers/loaders/single_file.py +++ b/src/diffusers/loaders/single_file.py @@ -282,7 +282,7 @@ class FromSingleFileMixin: ) if torch_dtype is not None: - pipe.to(torch_dtype=torch_dtype) + pipe.to(dtype=torch_dtype) return pipe