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,
|
||||
)
|
||||
vae_model.save_pretrained(output_path, safe_serialization=True)
|
||||
return output_path
|
||||
return vae_model
|
||||
|
Loading…
Reference in New Issue
Block a user