Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup logger

Browse Source
This commit is contained in:
Vladimir Mandic
2026-02-19 11:09:13 +01:00
parent 22aa9eab01
commit e5c494f999
279 changed files with 2898 additions and 2771 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_callbacks.py
+5 -5
View File
@@ -3,12 +3,12 @@ import time
import torch
import numpy as np
from modules import shared, devices, processing_correction, timer, prompt_parser_diffusers
from modules import logger
from modules.logger import log
p = None
debug = os.environ.get('SD_CALLBACK_DEBUG', None) is not None
debug_callback = logger.log.trace if debug else lambda *args, **kwargs: None
debug_callback = log.trace if debug else lambda *args, **kwargs: None
warned = False
@@ -44,7 +44,7 @@ def diffusers_callback_legacy(step: int, timestep: int, latents: torch.FloatTens
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
if shared.state.paused:
logger.log.debug('Sampling paused')
log.debug('Sampling paused')
while shared.state.paused:
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
@@ -68,7 +68,7 @@ def diffusers_callback(pipe, step: int = 0, timestep: int = 0, kwargs: dict = No
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
if shared.state.paused:
logger.log.debug('Sampling paused')
log.debug('Sampling paused')
while shared.state.paused:
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
@@ -176,7 +176,7 @@ def diffusers_callback(pipe, step: int = 0, timestep: int = 0, kwargs: dict = No
except Exception as e:
global warned # pylint: disable=global-statement
if not warned:
logger.log.error(f'Callback: {e}')
log.error(f'Callback: {e}')
warned = True
# from modules import errors
# errors.display(e, 'Callback')