apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

resolve VAE files within model path

This commit is contained in:
Sean Sube 2023-02-16 18:56:54 -06:00
parent 2c66cc58c5
commit aa6dc334d7
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1311,6 +1311,7 @@ def extract_checkpoint(
if vae_file is None:
converted_vae_checkpoint = convert_ldm_vae_checkpoint(checkpoint, vae_config)
else:
vae_file = os.path.join(ctx.model_path, vae_file)
vae_checkpoint = safetensors.torch.load_file(vae_file, device="cpu")
converted_vae_checkpoint = convert_ldm_vae_checkpoint(vae_checkpoint, vae_config)