From ddd3d45b79f08d82055a143ea7492444705ab071 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 23 Nov 2023 23:21:45 -0600 Subject: [PATCH] fix(api): add logging along SDXL custom VAE codepath --- api/onnx_web/convert/diffusion/diffusion_xl.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/onnx_web/convert/diffusion/diffusion_xl.py b/api/onnx_web/convert/diffusion/diffusion_xl.py index a4f11dd9..7a03b700 100644 --- a/api/onnx_web/convert/diffusion/diffusion_xl.py +++ b/api/onnx_web/convert/diffusion/diffusion_xl.py @@ -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):