apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-03-18 11:50:09 -05:00
parent a1f17c0877
commit e104c81e19
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/convert/__main__.py
View File

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

12
api/onnx_web/convert/diffusion/textual_inversion.py
View File

@ -145,9 +145,9 @@ def convert_diffusion_textual_inversion(
name: str,
base_model: str,
inversion: str,
format: str,
inversion_format: str,
base_token: Optional[str] = None,
weight: Optional[float] = 1.0,
inversion_weight: Optional[float] = 1.0,
):
dest_path = path.join(context.model_path, f"inversion-{name}")
logger.info(
@ -177,16 +177,16 @@ def convert_diffusion_textual_inversion(
context,
text_encoder,
tokenizer,
[(inversion, weight, base_token, format)],
[(inversion, inversion_weight, base_token, inversion_format)],
)
logger.info("saving tokenizer for Textual Inversion")
logger.info("saving tokenizer for textual inversion")
tokenizer.save_pretrained(tokenizer_path)
logger.info("saving text encoder for Textual Inversion")
logger.info("saving text encoder for textual inversion")
save_model(
text_encoder,
f=encoder_model,
)
logger.info("Textual Inversion saved to %s", dest_path)
logger.info("textual inversion saved to %s", dest_path)