jadechoghari
/

mar

jadechoghari commited on Sep 11

Commit

5bc55c8

•

1 Parent(s): c8f761f

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -79,7 +79,7 @@ class MARModel(DiffusionPipeline):
         # download and load the vae
         vae_checkpoint_path = hf_hub_download(
             repo_id=kwargs.get("repo_id", "jadechoghari/mar"),
-            filename=kwargs.get("vae_filename", "kl16.ckpt")
         )
         vae_checkpoint_path = kwargs.get("vae_checkpoint_path", vae_checkpoint_path)

         # download and load the vae
         vae_checkpoint_path = hf_hub_download(
             repo_id=kwargs.get("repo_id", "jadechoghari/mar"),
+            filename=kwargs.get("vae_filename", "kl16.safetensors")
         )
         vae_checkpoint_path = kwargs.get("vae_checkpoint_path", vae_checkpoint_path)