feat(mm): use same pattern for vae converter as others

Add `dump_path` arg to the converter function & save the model to disk inside the conversion function. This is the same pattern as in the other conversion functions.
This commit is contained in:
psychedelicious
2024-04-01 12:24:58 +11:00
parent 13f410478a
commit 59b4a23479
2 changed files with 11 additions and 5 deletions

View File

@ -64,6 +64,6 @@ class VAELoader(GenericDiffusersLoader):
vae_config=ckpt_config,
image_size=512,
precision=self._torch_dtype,
dump_path=output_path,
)
vae_model.save_pretrained(output_path, safe_serialization=True)
return vae_model