From 077845e8682e3427acf6021319937ae040e21aa7 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 29 Apr 2023 23:11:46 -0500 Subject: [PATCH] fix log levels --- api/onnx_web/convert/diffusion/diffusers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/convert/diffusion/diffusers.py b/api/onnx_web/convert/diffusion/diffusers.py index bf3fc8f8..85de6362 100644 --- a/api/onnx_web/convert/diffusion/diffusers.py +++ b/api/onnx_web/convert/diffusion/diffusers.py @@ -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: