apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): avoid passing cache path before sanitizing config name

This commit is contained in:
Sean Sube 2023-05-27 12:27:33 -05:00
parent 37fcd41c9b
commit 86d458a0c6
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 2 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

@ -1688,7 +1688,7 @@ def convert_extract_checkpoint(
config_file: Optional[str] = None,
vae_file: Optional[str] = None,
) -> Tuple[bool, str]:
working_name = os.path.join(dest, "working")
working_name = os.path.join(conversion.cache_path, dest, "working")
model_index = os.path.join(working_name, "model_index.json")
if os.path.exists(working_name) and os.path.exists(model_index):

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

@ -333,7 +333,7 @@ def convert_diffusion_diffusers(
torch_source = convert_extract_checkpoint(
conversion,
source,
path.join(conversion.cache_path, name + "-torch"),
f"{name}-torch",
config_file=config,
vae_file=replace_vae,
)