diff --git a/comfy/model_management.py b/comfy/model_management.py index f2ca36ca8..e0ccca1a8 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -551,6 +551,7 @@ 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: @@ -626,7 +627,6 @@ 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)))