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)