apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): add logging along SDXL custom VAE codepath

This commit is contained in:
Sean Sube 2023-11-23 23:21:45 -06:00
parent 51f0806c4a
commit ddd3d45b79
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/convert/diffusion/diffusion_xl.py
View File

@ -65,8 +65,10 @@ def convert_diffusion_diffusers_xl(
if replace_vae is not None:
vae_path = path.join(conversion.model_path, replace_vae)
if check_ext(replace_vae, RESOLVE_FORMATS):
logger.debug("loading VAE from single tensor file: %s", vae_path)
pipeline.vae = AutoencoderKL.from_single_file(vae_path)
else:
logger.debug("loading pretrained VAE from path: %s", vae_path)
pipeline.vae = AutoencoderKL.from_pretrained(vae_path)
if path.exists(temp_path):