apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

use torch suffix consistently

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

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

@ -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,