apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

go back to working dir

This commit is contained in:
Sean Sube 2023-12-24 06:10:15 -06:00
parent 769350115c
commit 525fec24c5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/convert/diffusion/checkpoint.py
View File

@ -178,7 +178,7 @@ class TrainingConfig:
): ):
model_name = sanitize_name(model_name) model_name = sanitize_name(model_name)
model_dir = os.path.join(conversion.cache_path, 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): if not os.path.exists(working_dir):
os.makedirs(working_dir) os.makedirs(working_dir)
@ -1693,7 +1693,7 @@ def convert_extract_checkpoint(
config_file: Optional[str] = None, config_file: Optional[str] = None,
vae_file: Optional[str] = None, vae_file: Optional[str] = None,
) -> Tuple[bool, str]: ) -> 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") model_index = os.path.join(dest_name, "model_index.json")
if os.path.exists(dest_name) and os.path.exists(model_index): if os.path.exists(dest_name) and os.path.exists(model_index):