mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-08 22:35:09 +00:00
31 lines
774 B
Python
31 lines
774 B
Python
import logging
|
|
import sys
|
|
import threading
|
|
from typing import *
|
|
|
|
|
|
def fire_in_thread(f, *args, **kwargs):
|
|
threading.Thread(target=f, args=args, kwargs=kwargs).start()
|
|
|
|
|
|
def setup_logging(log_level=logging.INFO):
|
|
if logging.root.handlers: # Already configured
|
|
return
|
|
|
|
try:
|
|
from rich.logging import RichHandler
|
|
|
|
handler = RichHandler()
|
|
except ImportError:
|
|
print("rich is not installed, using basic logging")
|
|
handler = logging.StreamHandler(sys.stdout) # same as print
|
|
handler.propagate = False
|
|
|
|
formatter = logging.Formatter(
|
|
fmt="%(message)s",
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|
)
|
|
handler.setFormatter(formatter)
|
|
logging.root.setLevel(log_level)
|
|
logging.root.addHandler(handler)
|