From aa6dc334d7f80b7ea783a8fb0a39ab951a9862e1 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 16 Feb 2023 18:56:54 -0600 Subject: [PATCH] resolve VAE files within model path --- api/onnx_web/convert/diffusion_original.py | 1 + 1 file changed, 1 insertion(+) diff --git a/api/onnx_web/convert/diffusion_original.py b/api/onnx_web/convert/diffusion_original.py index 602e698e..4fae214b 100644 --- a/api/onnx_web/convert/diffusion_original.py +++ b/api/onnx_web/convert/diffusion_original.py @@ -1311,6 +1311,7 @@ def extract_checkpoint( if vae_file is None: converted_vae_checkpoint = convert_ldm_vae_checkpoint(checkpoint, vae_config) else: + vae_file = os.path.join(ctx.model_path, vae_file) vae_checkpoint = safetensors.torch.load_file(vae_file, device="cpu") converted_vae_checkpoint = convert_ldm_vae_checkpoint(vae_checkpoint, vae_config)