fix(api): pass replacement VAE to diffusers conversion

2023-08-20 15:07:17 -05:00 · 2023-08-20 15:07:17 -05:00 · 8e1f188d8f
parent 58795de00d
commit 8e1f188d8f
1 changed files with 1 additions and 0 deletions
--- a/api/onnx_web/convert/diffusion/diffusers.py
+++ b/api/onnx_web/convert/diffusion/diffusers.py
@ -360,6 +360,7 @@ def convert_diffusion_diffusers(
                source,
                original_config_file=config_path,
                pipeline_class=pipe_class,
+                vae_path=replace_vae,
                **pipe_args,
            ).to(device, torch_dtype=dtype)
    elif hf: