apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

lint(api): move some noisy log messages to trace level

This commit is contained in:
Sean Sube 2023-12-21 21:07:02 -06:00
parent a1fc60641f
commit 2546084208
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -510,7 +510,7 @@ def convert_diffusion_diffusers(
run_gc()
# UNET
logger.debug("UNET config: %s", pipeline.unet.config)
logger.trace("UNET config: %s", pipeline.unet.config)
if single_vae:
unet_inputs = ["sample", "timestep", "encoder_hidden_states", "class_labels"]
unet_scale = torch.tensor(4).to(device=device, dtype=torch.long)

4
api/onnx_web/server/load.py
View File

@ -502,7 +502,7 @@ def load_wildcards(server: ServerContext) -> None:
lines = f.read().splitlines()
lines = [line.strip() for line in lines if not line.startswith("#")]
lines = [line for line in lines if len(line) > 0]
logger.debug("loading wildcards from %s: %s", file, lines)
logger.trace("loading wildcards from %s: %s", file, lines)
wildcard_data[path.splitext(file)[0]].extend(lines)
structured_files = list_model_globs(
@ -515,7 +515,7 @@ def load_wildcards(server: ServerContext) -> None:
for file in structured_files:
data = load_config(path.join(wildcard_path, file))
logger.debug("loading structured wildcards from %s: %s", file, data)
logger.trace("loading structured wildcards from %s: %s", file, data)
parse_wildcards(data, root_key=path.splitext(file)[0])