fix(api): only convert VAE once when coming from checkpoints

ssube · Feb 17, 2023 · 7a1f831 · 7a1f831
1 parent 37b173d
commit 7a1f831
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/onnx_web/convert/diffusion_original.py b/api/onnx_web/convert/diffusion_original.py
@@ -1427,5 +1427,8 @@ def convert_diffusion_original(
         extract_checkpoint(ctx, torch_name, source, config_file=model.get("config"), vae_file=model.get("vae"))
         logger.info("converted original Diffusers checkpoint to Torch model")
 
+    # VAE has already been converted and will confuse HF repo lookup
+    del model["vae"]
+
     convert_diffusion_stable(ctx, model, working_name)
     logger.info("ONNX pipeline saved to %s", name)