mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
fix(mm): return converted vae model instead of path
This was missed in #6072.
This commit is contained in:
parent
f83edcf990
commit
25ff0bf80f
@ -65,4 +65,4 @@ class VAELoader(GenericDiffusersLoader):
|
|||||||
precision=self._torch_dtype,
|
precision=self._torch_dtype,
|
||||||
)
|
)
|
||||||
vae_model.save_pretrained(output_path, safe_serialization=True)
|
vae_model.save_pretrained(output_path, safe_serialization=True)
|
||||||
return output_path
|
return vae_model
|
||||||
|
Loading…
x
Reference in New Issue
Block a user