aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAarni Koskela <akx@iki.fi>2023-06-30 10:44:58 +0000
committerAarni Koskela <akx@iki.fi>2023-07-03 08:03:41 +0000
commitb70001e618d0f0015273e1313cc7ebe3002a4510 (patch)
tree31d4644a27ade697a5a97636157db90fdbd60de7
parentfab73f2e7d388ca99cdb3d5de7f36c0b9a1a3b1c (diff)
downloadstable-diffusion-webui-gfx803-b70001e618d0f0015273e1313cc7ebe3002a4510.tar.gz
stable-diffusion-webui-gfx803-b70001e618d0f0015273e1313cc7ebe3002a4510.tar.bz2
stable-diffusion-webui-gfx803-b70001e618d0f0015273e1313cc7ebe3002a4510.zip
Add SD_WEBUI_LOG_LEVEL envvar
-rw-r--r--webui.py11
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