apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix dest paths for blended nets

This commit is contained in:
Sean Sube 2023-03-18 11:37:16 -05:00
parent f465120cad
commit a1f17c0877
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/__main__.py
View File

@ -348,14 +348,14 @@ def convert_models(ctx: ConversionContext, args, models: Models):
)
if "tokenizer" in blend_models:
dest_path = path.join(ctx.model_path, model, "tokenizer")
dest_path = path.join(dest, "tokenizer")
logger.debug("saving blended tokenizer to %s", dest_path)
blend_models["tokenizer"].save_pretrained(dest_path)
for name in ["text_encoder", "unet"]:
if name in blend_models:
dest_path = path.join(
ctx.model_path, model, name, "model.onnx"
dest, name, "model.onnx"
)
logger.debug(
"saving blended %s model to %s", name, dest_path