use dest in file client if provided

ssube · Dec 10, 2023 · c9b1df9 · c9b1df9
1 parent e91e084
commit c9b1df9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/api/onnx_web/convert/client/civitai.py b/api/onnx_web/convert/client/civitai.py
@@ -26,7 +26,7 @@ def __init__(
         self,
         conversion: ConversionContext,
         token: Optional[str] = None,
-        root=CIVITAI_ROOT,
+        root: str = CIVITAI_ROOT,
     ):
         self.root = conversion.get_setting("CIVITAI_ROOT", root)
         self.token = conversion.get_setting("CIVITAI_TOKEN", token)

diff --git a/api/onnx_web/convert/client/file.py b/api/onnx_web/convert/client/file.py
@@ -29,4 +29,4 @@ def download(
     ) -> str:
         parts = urlparse(uri)
         logger.info("loading model from: %s", parts.path)
-        return path.join(conversion.model_path, parts.path)
+        return path.join(dest or conversion.model_path, parts.path)