apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): skip loading custom VAE after torch extraction (#379)

This commit is contained in:
Sean Sube 2023-05-27 12:31:08 -05:00
parent 3adafeaa3d
commit 00b1bec7a9
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/diffusers.py
View File

@ -344,6 +344,9 @@ def convert_diffusion_diffusers(
torch_source, torch_source,
torch_dtype=dtype, torch_dtype=dtype,
).to(device) ).to(device)
# VAE replacement already happened during extraction, skip
replace_vae = None
else: else:
logger.debug("loading pipeline from SD checkpoint: %s", source) logger.debug("loading pipeline from SD checkpoint: %s", source)
pipeline = download_from_original_stable_diffusion_ckpt( pipeline = download_from_original_stable_diffusion_ckpt(