apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): only convert VAE once when coming from checkpoints

This commit is contained in:
Sean Sube 2023-02-17 08:29:23 -06:00
parent 37b173d0d1
commit 7a1f8318ef
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/convert/diffusion_original.py
View File

@ -1427,5 +1427,8 @@ def convert_diffusion_original(
extract_checkpoint(ctx, torch_name, source, config_file=model.get("config"), vae_file=model.get("vae")) 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") 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) convert_diffusion_stable(ctx, model, working_name)
logger.info("ONNX pipeline saved to %s", name) logger.info("ONNX pipeline saved to %s", name)