From e7da2cf8a6d0a3d6d455ac4248453a55aa9d7034 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 8 Dec 2023 23:19:52 -0600 Subject: [PATCH] fix(api): load pretrained VAE from original path --- api/onnx_web/convert/diffusion/diffusion.py | 4 ++-- api/onnx_web/convert/diffusion/diffusion_xl.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py index ade270c0..dbb50610 100644 --- a/api/onnx_web/convert/diffusion/diffusion.py +++ b/api/onnx_web/convert/diffusion/diffusion.py @@ -381,11 +381,11 @@ def convert_diffusion_diffusers( if replace_vae is not None: vae_path = path.join(conversion.model_path, replace_vae) - vae_file = check_ext(replace_vae, RESOLVE_FORMATS) + vae_file = check_ext(vae_path, RESOLVE_FORMATS) if vae_file[0]: pipeline.vae = AutoencoderKL.from_single_file(vae_path) else: - pipeline.vae = AutoencoderKL.from_pretrained(vae_path) + pipeline.vae = AutoencoderKL.from_pretrained(replace_vae) if is_torch_2_0: pipeline.unet.set_attn_processor(AttnProcessor()) diff --git a/api/onnx_web/convert/diffusion/diffusion_xl.py b/api/onnx_web/convert/diffusion/diffusion_xl.py index 26f50bd8..8370d302 100644 --- a/api/onnx_web/convert/diffusion/diffusion_xl.py +++ b/api/onnx_web/convert/diffusion/diffusion_xl.py @@ -69,8 +69,8 @@ def convert_diffusion_diffusers_xl( 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) + logger.debug("loading pretrained VAE from path: %s", replace_vae) + pipeline.vae = AutoencoderKL.from_pretrained(replace_vae) if path.exists(temp_path): logger.debug("torch model already exists for %s: %s", source, temp_path)