apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-05-28 19:44:16 -05:00
parent 2a9f3a165b
commit 4dc192a2c7
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 5 additions and 5 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

@ -51,7 +51,7 @@ from transformers import (
CLIPVisionConfig, CLIPVisionConfig,
) )
from ...utils import sanitize_name, load_config from ...utils import load_config, sanitize_name
from ..utils import ConversionContext, load_tensor from ..utils import ConversionContext, load_tensor
logger = getLogger(__name__) logger = getLogger(__name__)

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

@ -337,9 +337,7 @@ def convert_diffusion_diffusers(
config_file=config, config_file=config,
vae_file=replace_vae, vae_file=replace_vae,
) )
logger.debug( logger.debug("loading pipeline from extracted checkpoint: %s", torch_source)
"loading pipeline from extracted checkpoint: %s", torch_source
)
pipeline = pipe_class.from_pretrained( pipeline = pipe_class.from_pretrained(
torch_source, torch_source,
torch_dtype=dtype, torch_dtype=dtype,
@ -521,7 +519,9 @@ def convert_diffusion_diffusers(
# VAE # VAE
if replace_vae is not None: if replace_vae is not None:
if replace_vae.startswith("."): if replace_vae.startswith("."):
logger.debug("custom VAE appears to be a local path, making it relative to the model path") logger.debug(
"custom VAE appears to be a local path, making it relative to the model path"
)
replace_vae = path.join(conversion.model_path, replace_vae) replace_vae = path.join(conversion.model_path, replace_vae)
logger.info("loading custom VAE: %s", replace_vae) logger.info("loading custom VAE: %s", replace_vae)