fix(api): only convert VAE once when coming from checkpoints
This commit is contained in:
parent
37b173d0d1
commit
7a1f8318ef
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue