apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix log levels

This commit is contained in:
Sean Sube 2023-04-29 23:11:46 -05:00
parent 62abc76f3f
commit 077845e868
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/diffusion/diffusers.py
View File

@ -69,7 +69,7 @@ def get_model_version(
if "global_step" in checkpoint:
global_step = checkpoint["global_step"]
else:
print("global_step key not found in model")
logger.trace("global_step key not found in model")
global_step = None
if size is None:
@ -84,7 +84,7 @@ def get_model_version(
v2 = True
if size != 512:
# v2.1 needs to upcast attention
logger.debug("setting upcast_attention")
logger.trace("setting upcast_attention")
opts["upcast_attention"] = True
if v2 and size != 512: