From 525fec24c52a306220593725e1688ce2b3110a17 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 24 Dec 2023 06:10:15 -0600 Subject: [PATCH] go back to working dir --- api/onnx_web/convert/diffusion/checkpoint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/convert/diffusion/checkpoint.py b/api/onnx_web/convert/diffusion/checkpoint.py index a486d3f9..ce49469a 100644 --- a/api/onnx_web/convert/diffusion/checkpoint.py +++ b/api/onnx_web/convert/diffusion/checkpoint.py @@ -178,7 +178,7 @@ class TrainingConfig: ): model_name = sanitize_name(model_name) model_dir = os.path.join(conversion.cache_path, model_name) - working_dir = f"{model_dir}-torch" + working_dir = os.path.join(model_dir, "working") if not os.path.exists(working_dir): os.makedirs(working_dir) @@ -1693,7 +1693,7 @@ 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") + dest_name = os.path.join(conversion.cache_path, name, "working") model_index = os.path.join(dest_name, "model_index.json") if os.path.exists(dest_name) and os.path.exists(model_index):