[From single file] remove depr warning (#6043)

huggingface · Dec 5, 2023 · 53bc30d · 53bc30d
1 parent eacf5e3
commit 53bc30d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/loaders/single_file.py b/src/diffusers/loaders/single_file.py
@@ -282,7 +282,7 @@ def from_single_file(cls, pretrained_model_link_or_path, **kwargs):
         )
 
         if torch_dtype is not None:
-            pipe.to(torch_dtype=torch_dtype)
+            pipe.to(dtype=torch_dtype)
 
         return pipe