fix(api): use correct post-fetch path when converting from checkpoints (#432)

2023-12-16 13:17:24 -06:00 · 2023-12-16 13:17:24 -06:00 · 133c4a20bd
parent 5b382afc6c
commit 133c4a20bd
1 changed files with 1 additions and 1 deletions
--- a/api/onnx_web/convert/diffusion/diffusion.py
+++ b/api/onnx_web/convert/diffusion/diffusion.py
@ -370,7 +370,7 @@ def convert_diffusion_diffusers(
            else:
                logger.debug("loading pipeline from SD checkpoint: %s", source)
                pipeline = download_from_original_stable_diffusion_ckpt(
-                    source,
+                    cache_path,
                    original_config_file=config_path,
                    pipeline_class=pipe_class,
                    **pipe_args,