From 0d15a091c25dbcbd56485f48c56a1df68aebcd7a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rafa=C5=82=20Leszko?= Date: Tue, 4 Feb 2025 14:44:12 +0100 Subject: [PATCH] Update logging when models are loaded The "Loaded " log was logged even if no model were actually loaded into VRAM --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index e0ccca1a8..f2ca36ca8 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -551,7 +551,6 @@ def load_models_gpu(models: Sequence[ModelManageable], memory_required: int = 0, _load_models_gpu(models, memory_required, force_patch_weights, minimum_memory_required, force_full_load) to_load = list(map(str, models)) span.set_attribute("models", to_load) - logger.info(f"Loaded {to_load}") def _load_models_gpu(models: Sequence[ModelManageable], memory_required: int = 0, force_patch_weights=False, minimum_memory_required=None, force_full_load=False) -> None: @@ -627,6 +626,7 @@ def _load_models_gpu(models: Sequence[ModelManageable], memory_required: int = 0 loaded_model.model_load(lowvram_model_memory, force_patch_weights=force_patch_weights) current_loaded_models.insert(0, loaded_model) + logger.info(f"Loaded {loaded_model}") span = get_current_span() span.set_attribute("models_to_load", list(map(str, models_to_load)))