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

.

[Bug Fix] Should pass the dtype instead of torch_dtype (#4917)
.
d82157b3 · zhiqiang · GitHub · 93579650 · d82157b3
Unverified Commit d82157b3 authored Sep 12, 2023 by zhiqiang Committed by GitHub Sep 11, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/loaders.py src/diffusers/loaders.py +1 -1

No files found.
--- a/src/diffusers/loaders.py
+++ b/src/diffusers/loaders.py
@@ -2381,7 +2381,7 @@ class FromOriginalVAEMixin:
            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