Skip to content

GitLab

Unverified Commit d82157b3 authored by zhiqiang's avatar zhiqiang Committed by GitHub
Browse files

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

.
parent 93579650
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/loaders.py
View file @ d82157b3
...@@ -2381,7 +2381,7 @@ class FromOriginalVAEMixin: ...@@ -2381,7 +2381,7 @@ class FromOriginalVAEMixin:
vae.load_state_dict(converted_vae_checkpoint) vae.load_state_dict(converted_vae_checkpoint)
if torch_dtype is not None: if torch_dtype is not None:
vae.to(torch_dtype=torch_dtype) vae.to(dtype=torch_dtype)
return vae return vae
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment