From 7a1f8318efc5b90e4d9d4d873470419324386f4c Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 17 Feb 2023 08:29:23 -0600 Subject: [PATCH] fix(api): only convert VAE once when coming from checkpoints --- api/onnx_web/convert/diffusion_original.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/api/onnx_web/convert/diffusion_original.py b/api/onnx_web/convert/diffusion_original.py index b11d1458..b0637d87 100644 --- 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)