apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

add torch suffix to temp dir

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

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

@ -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, name, "working")
dest_name = os.path.join(conversion.cache_path, f"{name}-torch", "working")
model_index = os.path.join(dest_name, "model_index.json")
if os.path.exists(dest_name) and os.path.exists(model_index):