apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

use dest in file client if provided

This commit is contained in:
Sean Sube 2023-12-10 13:30:27 -06:00
parent e91e08484b
commit c9b1df9fdd
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/convert/client/civitai.py
View File

@ -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)

2
api/onnx_web/convert/client/file.py
View File

@ -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)