[Bug Fix] Should pass the dtype instead of torch_dtype (huggingface#4917)

zhiqiang-canva · web-flow · commit d82157b3ce78 · 2023-09-11T19:45:58.000+02:00
.
diff --git a/src/diffusers/loaders.py b/src/diffusers/loaders.py
@@ -2381,7 +2381,7 @@ def from_single_file(cls, pretrained_model_link_or_path, **kwargs):
             vae.load_state_dict(converted_vae_checkpoint)
 
         if torch_dtype is not None:
-            vae.to(torch_dtype=torch_dtype)
+            vae.to(dtype=torch_dtype)
 
         return vae
 
