From 56a826228d93f6919c67806437f870e6010dec6a Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 17 Feb 2023 08:34:28 -0600 Subject: [PATCH] fix(api): only remove converted VAE if it exists --- api/onnx_web/convert/diffusion_original.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/convert/diffusion_original.py b/api/onnx_web/convert/diffusion_original.py index b0637d87..45dd0ada 100644 --- 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)