From 86d458a0c675aa7ae87526a77d5edad1134cae1b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 27 May 2023 12:27:33 -0500 Subject: [PATCH] fix(api): avoid passing cache path before sanitizing config name --- api/onnx_web/convert/diffusion/checkpoint.py | 2 +- api/onnx_web/convert/diffusion/diffusers.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/convert/diffusion/checkpoint.py b/api/onnx_web/convert/diffusion/checkpoint.py index 880167af1..91aed48b3 100644 --- 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 index 7b17507b3..14fc4dc37 100644 --- 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, )