wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 14:50:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Merge pull request #28 from leszko/patch-2

Browse Source 
Update logging when models are loaded
This commit is contained in:
Benjamin Berman 2025-02-04 08:28:30 -08:00 committed by GitHub
commit fac670da89
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -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)))