diff --git a/comfy/model_management.py b/comfy/model_management.py index 1f5b9ddb8..2c702959a 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -594,11 +594,11 @@ def load_models_gpu(models, memory_required=0, force_patch_weights=False, minimu inference_memory = minimum_inference_memory() models_memory_reserve = get_models_memory_reserve(models) - extra_mem = max(inference_memory, memory_required + extra_reserved_memory() + models_memory_reserve) + extra_mem = max(inference_memory + models_memory_reserve, memory_required + extra_reserved_memory() + models_memory_reserve) if minimum_memory_required is None: minimum_memory_required = extra_mem else: - minimum_memory_required = max(inference_memory, minimum_memory_required + extra_reserved_memory() + models_memory_reserve) + minimum_memory_required = max(inference_memory + models_memory_reserve, minimum_memory_required + extra_reserved_memory() + models_memory_reserve) models = set(models)