Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-26 15:16:01 +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/transformer_cache.py
+6 -5
View File
@@ -1,9 +1,10 @@
import os
import diffusers
from modules import shared, errors
from modules import logger
debug = shared.log.trace if os.environ.get('SD_PROCESS_DEBUG', None) is not None else lambda *args, **kwargs: None
debug = logger.log.trace if os.environ.get('SD_PROCESS_DEBUG', None) is not None else lambda *args, **kwargs: None
def set_cache(faster_cache=None, pyramid_attention_broadcast=None):
@@ -14,7 +15,7 @@ def set_cache(faster_cache=None, pyramid_attention_broadcast=None):
if (not faster_cache) and (not pyramid_attention_broadcast):
return
if (not hasattr(shared.sd_model.transformer, 'enable_cache')) or (not hasattr(shared.sd_model.transformer, 'disable_cache')):
shared.log.debug(f'Transformer cache: cls={shared.sd_model.transformer.__class__.__name__} fc={faster_cache} pab={pyramid_attention_broadcast} not supported')
logger.log.debug(f'Transformer cache: cls={shared.sd_model.transformer.__class__.__name__} fc={faster_cache} pab={pyramid_attention_broadcast} not supported')
return
try:
if faster_cache: # https://github.com/huggingface/diffusers/pull/10163
@@ -31,7 +32,7 @@ def set_cache(faster_cache=None, pyramid_attention_broadcast=None):
)
shared.sd_model.transformer.disable_cache()
shared.sd_model.transformer.enable_cache(config)
shared.log.debug(f'Transformer cache: type={config.__class__.__name__}')
logger.log.debug(f'Transformer cache: type={config.__class__.__name__}')
debug(f'Transformer cache: {vars(config)}')
elif pyramid_attention_broadcast: # https://github.com/huggingface/diffusers/pull/9562
config = diffusers.PyramidAttentionBroadcastConfig(
@@ -41,11 +42,11 @@ def set_cache(faster_cache=None, pyramid_attention_broadcast=None):
)
shared.sd_model.transformer.disable_cache()
shared.sd_model.transformer.enable_cache(config)
shared.log.debug(f'Transformer cache: type={config.__class__.__name__}')
logger.log.debug(f'Transformer cache: type={config.__class__.__name__}')
debug(f'Transformer cache: {vars(config)}')
else:
debug('Transformer cache: not enabled')
shared.sd_model.transformer.disable_cache()
except Exception as e:
shared.log.error(f'Transformer cache: {e}')
logger.log.error(f'Transformer cache: {e}')
errors.display(e, 'Transformer cache')