fix(api): avoid passing cache path before sanitizing config name

ssube · May 27, 2023 · 86d458a · 86d458a
1 parent 37fcd41
commit 86d458a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/api/onnx_web/convert/diffusion/checkpoint.py b/api/onnx_web/convert/diffusion/checkpoint.py
@@ -1688,7 +1688,7 @@ def convert_extract_checkpoint(
     config_file: Optional[str] = None,
     vae_file: Optional[str] = None,
 ) -> Tuple[bool, str]:
-    working_name = os.path.join(dest, "working")
+    working_name = os.path.join(conversion.cache_path, dest, "working")
     model_index = os.path.join(working_name, "model_index.json")
 
     if os.path.exists(working_name) and os.path.exists(model_index):

diff --git a/api/onnx_web/convert/diffusion/diffusers.py b/api/onnx_web/convert/diffusion/diffusers.py
@@ -333,7 +333,7 @@ def convert_diffusion_diffusers(
             torch_source = convert_extract_checkpoint(
                 conversion,
                 source,
-                path.join(conversion.cache_path, name + "-torch"),
+                f"{name}-torch",
                 config_file=config,
                 vae_file=replace_vae,
             )