Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

unified logger

Browse Source
This commit is contained in:
Vladimir Mandic
2026-02-19 09:46:42 +01:00
parent bfe014f5da
commit a3074baf8b
315 changed files with 2507 additions and 2116 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_callbacks.py
+5 -4
View File
@@ -3,11 +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
p = None
debug = os.environ.get('SD_CALLBACK_DEBUG', None) is not None
debug_callback = shared.log.trace if debug else lambda *args, **kwargs: None
debug_callback = logger.log.trace if debug else lambda *args, **kwargs: None
warned = False
@@ -43,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:
shared.log.debug('Sampling paused')
logger.log.debug('Sampling paused')
while shared.state.paused:
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
@@ -67,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:
shared.log.debug('Sampling paused')
logger.log.debug('Sampling paused')
while shared.state.paused:
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
@@ -175,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:
shared.log.error(f'Callback: {e}')
logger.log.error(f'Callback: {e}')
warned = True
# from modules import errors
# errors.display(e, 'Callback')