From c9b1df9fdd0517b33230cac9b0e271ef66da3404 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 10 Dec 2023 13:30:27 -0600 Subject: [PATCH] use dest in file client if provided --- api/onnx_web/convert/client/civitai.py | 2 +- api/onnx_web/convert/client/file.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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)