fix(api): pass replacement VAE to diffusers conversion

ssube · Sep 5, 2023 · 8e1f188 · 8e1f188
1 parent 58795de
commit 8e1f188
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: