Merge pull request #11593 from akx/better-status-reporting-1

Better status reporting, part 1

1 files changed, 11 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index bad29f28..1b44d4ad 100644
--- a/webui.py
+++ b/webui.py
@@ -18,6 +18,17 @@ from packaging import version
 

 import logging

 

+# We can't use cmd_opts for this because it will not have been initialized at this point.

+log_level = os.environ.get("SD_WEBUI_LOG_LEVEL")

+if log_level:

+    log_level = getattr(logging, log_level.upper(), None) or logging.INFO

+    logging.basicConfig(

+        level=log_level,

+        format='%(asctime)s %(levelname)s [%(name)s] %(message)s',

+        datefmt='%Y-%m-%d %H:%M:%S',

+    )

+

+logging.getLogger("torch.distributed.nn").setLevel(logging.ERROR)  # sshh...

 logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())

 

 from modules import paths, timer, import_hook, errors, devices  # noqa: F401