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"]: for name in ["text_encoder", "unet"]:
if name in blend_models: if name in blend_models:
dest_path = path.join( dest_path = path.join(dest, name, "model.onnx")
dest, name, "model.onnx"
)
logger.debug( logger.debug(
"saving blended %s model to %s", name, dest_path "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, name: str,
base_model: str, base_model: str,
inversion: str, inversion: str,
format: str, inversion_format: str,
base_token: Optional[str] = None, 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}") dest_path = path.join(context.model_path, f"inversion-{name}")
logger.info( logger.info(
@ -177,16 +177,16 @@ def convert_diffusion_textual_inversion(
context, context,
text_encoder, text_encoder,
tokenizer, 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) tokenizer.save_pretrained(tokenizer_path)
logger.info("saving text encoder for Textual Inversion") logger.info("saving text encoder for textual inversion")
save_model( save_model(
text_encoder, text_encoder,
f=encoder_model, f=encoder_model,
) )
logger.info("Textual Inversion saved to %s", dest_path) logger.info("textual inversion saved to %s", dest_path)