fix(api): only remove converted VAE if it exists

ssube · Feb 17, 2023 · 56a8262 · 56a8262
1 parent 7a1f831
commit 56a8262
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/onnx_web/convert/diffusion_original.py b/api/onnx_web/convert/diffusion_original.py
@@ -1428,7 +1428,8 @@ def convert_diffusion_original(
         logger.info("converted original Diffusers checkpoint to Torch model")
 
     # VAE has already been converted and will confuse HF repo lookup
-    del model["vae"]
+    if "vae" in model:
+        del model["vae"]
 
     convert_diffusion_stable(ctx, model, working_name)
     logger.info("ONNX pipeline saved to %s", name)