From 1735d4fb014070924726e372b560a77dd7592590 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Fri, 1 Nov 2024 04:25:27 -0400 Subject: [PATCH] Fix issue. --- comfy/model_management.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index bad3485d4..960be9785 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -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__}") models_to_load.append(loaded_model) - to_unload = [] for loaded_model in models_to_load: + to_unload = [] for i in range(len(current_loaded_models)): if loaded_model.model.is_clone(current_loaded_models[i].model): to_unload = [i] + to_unload - for i in to_unload: - current_loaded_models.pop(i).model.detach(unpatch_all=False) + for i in to_unload: + current_loaded_models.pop(i).model.detach(unpatch_all=False) total_memory_required = {} for loaded_model in models_to_load: