VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-06 13:47:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

update log initialization without rich

Browse Source
This commit is contained in:
Kohya S
2024-02-08 21:06:39 +09:00
parent efd3b58973
commit 9b8ea12d34
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
library/utils.py
View File

@@ -34,12 +34,14 @@ def setup_logging(args=None, log_level=None, reset=False):
else:
return
# log_level can be set by the caller or by the args, the caller has priority. If not set, use INFO
if log_level is None and args is not None:
log_level = args.console_log_level
if log_level is None:
log_level = "INFO"
log_level = getattr(logging, log_level)
msg_init = None
if args is not None and args.console_log_file:
handler = logging.FileHandler(args.console_log_file, mode="w")
else:
@@ -50,7 +52,8 @@ def setup_logging(args=None, log_level=None, reset=False):
handler = RichHandler()
except ImportError:
print("rich is not installed, using basic logging")
# print("rich is not installed, using basic logging")
msg_init = "rich is not installed, using basic logging"
if handler is None:
handler = logging.StreamHandler(sys.stdout) # same as print
@@ -63,3 +66,7 @@ def setup_logging(args=None, log_level=None, reset=False):
handler.setFormatter(formatter)
logging.root.setLevel(log_level)
logging.root.addHandler(handler)
if msg_init is not None:
logger = logging.getLogger(__name__)
logger.info(msg_init)