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

Fix issue.

Browse Source
This commit is contained in:
comfyanonymous 2024-11-01 04:25:27 -04:00
parent d8bd2a9baa
commit 1735d4fb01
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/model_management.py
View File

@ -496,13 +496,13 @@ def load_models_gpu(models, memory_required=0, force_patch_weights=False, minimu
logging.info(f"Requested to load {x.model.__class__.__name__}") logging.info(f"Requested to load {x.model.__class__.__name__}")
models_to_load.append(loaded_model) models_to_load.append(loaded_model)
to_unload = []
for loaded_model in models_to_load: for loaded_model in models_to_load:
to_unload = []
for i in range(len(current_loaded_models)): for i in range(len(current_loaded_models)):
if loaded_model.model.is_clone(current_loaded_models[i].model): if loaded_model.model.is_clone(current_loaded_models[i].model):
to_unload = [i] + to_unload to_unload = [i] + to_unload
for i in to_unload: for i in to_unload:
current_loaded_models.pop(i).model.detach(unpatch_all=False) current_loaded_models.pop(i).model.detach(unpatch_all=False)
total_memory_required = {} total_memory_required = {}
for loaded_model in models_to_load: for loaded_model in models_to_load: