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):