From ff4d57814e2291c80fa694dcd0508a17ea4da12e Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Fri, 4 Apr 2025 09:14:48 -0400 Subject: [PATCH] fix debug logging Signed-off-by: Vladimir Mandic --- modules/sd_models.py | 5 +++-- modules/sd_offload.py | 3 ++- modules/ui_common.py | 1 - 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 4e85ec74c..d4fd53a0b 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -9,6 +9,7 @@ import diffusers import diffusers.loaders.single_file_utils import torch +from installer import log from modules import paths, shared, shared_state, modelloader, devices, script_callbacks, sd_vae, sd_unet, errors, sd_models_config, sd_models_compile, sd_hijack_accelerate, sd_detect, model_quant from modules.timer import Timer, process as process_timer from modules.memstats import memory_stats @@ -25,9 +26,9 @@ sd_metadata_file = os.path.join(paths.data_path, "metadata.json") sd_metadata = None sd_metadata_pending = 0 sd_metadata_timer = 0 -debug_move = shared.log.trace if os.environ.get('SD_MOVE_DEBUG', None) is not None else lambda *args, **kwargs: None +debug_move = log.trace if os.environ.get('SD_MOVE_DEBUG', None) is not None else lambda *args, **kwargs: None debug_load = os.environ.get('SD_LOAD_DEBUG', None) -debug_process = shared.log.trace if os.environ.get('SD_PROCESS_DEBUG', None) is not None else lambda *args, **kwargs: None +debug_process = log.trace if os.environ.get('SD_PROCESS_DEBUG', None) is not None else lambda *args, **kwargs: None diffusers_version = int(diffusers.__version__.split('.')[1]) checkpoint_tiles = checkpoint_titles # legacy compatibility diff --git a/modules/sd_offload.py b/modules/sd_offload.py index e4b24c17b..e27b70ac2 100644 --- a/modules/sd_offload.py +++ b/modules/sd_offload.py @@ -4,11 +4,12 @@ import time import inspect import torch import accelerate.hooks +from installer import log from modules import shared, devices, errors, model_quant from modules.timer import process as process_timer -debug_move = shared.log.trace if os.environ.get('SD_MOVE_DEBUG', None) is not None else lambda *args, **kwargs: None +debug_move = log.trace if os.environ.get('SD_MOVE_DEBUG', None) is not None else lambda *args, **kwargs: None should_offload = ['sc', 'sd3', 'f1', 'hunyuandit', 'auraflow', 'omnigen', 'cogview4'] offload_hook_instance = None balanced_offload_exclude = ['OmniGenPipeline', 'CogView4Pipeline'] diff --git a/modules/ui_common.py b/modules/ui_common.py index 84e0c888d..d0696461a 100644 --- a/modules/ui_common.py +++ b/modules/ui_common.py @@ -255,7 +255,6 @@ def create_output_panel(tabname, preview=True, prompt=None, height=None, transfe elem_id=f"{tabname}_gallery", elem_classes=["gallery_main"], ) - print('HERE', shared.opts.ui_columns) if prompt is not None: ui_sections.create_interrogate_button(tab=tabname, inputs=result_gallery, outputs=prompt)