918d168ae7
The logic has not been changed, only renaming, splitting and moving modules on a per-function basis. Existing code will be left in place for the time being to avoid breaking the training code, which is not subject to refactoring this time.
16 lines
327 B
Python
16 lines
327 B
Python
from loguru import logger
|
|
|
|
from style_bert_vits2.utils.stdout_wrapper import SAFE_STDOUT
|
|
|
|
|
|
# Remove all default handlers
|
|
logger.remove()
|
|
|
|
# Add a new handler
|
|
logger.add(
|
|
SAFE_STDOUT,
|
|
format = "<g>{time:MM-DD HH:mm:ss}</g> |<lvl>{level:^8}</lvl>| {file}:{line} | {message}",
|
|
backtrace = True,
|
|
diagnose = True,
|
|
)
|