diff --git a/api/onnx_web/convert/client/civitai.py b/api/onnx_web/convert/client/civitai.py index 6bbb1be5..b57b6533 100644 --- a/api/onnx_web/convert/client/civitai.py +++ b/api/onnx_web/convert/client/civitai.py @@ -26,7 +26,7 @@ class CivitaiClient(BaseClient): 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 index 63077774..8ed1a4d9 100644 --- a/api/onnx_web/convert/client/file.py +++ b/api/onnx_web/convert/client/file.py @@ -29,4 +29,4 @@ class FileClient(BaseClient): ) -> 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)