fix(mm): return converted vae model instead of path

This was missed in #6072.
This commit is contained in:
psychedelicious 2024-04-01 12:17:39 +11:00
parent f83edcf990
commit 25ff0bf80f

View File

@ -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