From 5d31e85709863489fd245a31b1187ce8ade87dcb Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 24 Dec 2023 06:16:37 -0600 Subject: [PATCH] use torch suffix consistently --- api/onnx_web/convert/diffusion/checkpoint.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/convert/diffusion/checkpoint.py b/api/onnx_web/convert/diffusion/checkpoint.py index ce3c3418..0362494a 100644 --- a/api/onnx_web/convert/diffusion/checkpoint.py +++ b/api/onnx_web/convert/diffusion/checkpoint.py @@ -1693,7 +1693,8 @@ def convert_extract_checkpoint( config_file: Optional[str] = None, vae_file: Optional[str] = None, ) -> Tuple[bool, str]: - dest_name = os.path.join(conversion.cache_path, f"{name}-torch", "working") + temp_name = f"{name}-torch" + dest_name = os.path.join(conversion.cache_path, temp_name, "working") model_index = os.path.join(dest_name, "model_index.json") if os.path.exists(dest_name) and os.path.exists(model_index): @@ -1706,7 +1707,7 @@ def convert_extract_checkpoint( ) if extract_checkpoint( conversion, - name, + temp_name, source, config_file=config_file, is_inpainting=is_inpainting,