From abbe30c49eca830d394290678ce9a45d8be294fc Mon Sep 17 00:00:00 2001 From: fromfirstbyte Date: Thu, 25 Jun 2026 06:06:26 +0300 Subject: [PATCH 1/2] Added support for --lowvram argument for 'mps' devices. If passed weights would not be loaded to VRAM completely. With modified GGUF loader it allows to keep memory-mapped weights on disk. --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index b15d08ba1..374f5503d 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -938,7 +938,7 @@ def load_models_gpu(models, memory_required=0, force_patch_weights=False, minimu if lowvram_model_memory == 0: lowvram_model_memory = 0.1 - if vram_set_state == VRAMState.NO_VRAM: + if vram_set_state == VRAMState.NO_VRAM or (set_vram_to == VRAMState.LOW_VRAM and vram_state == VRAMState.SHARED and not force_full_load): lowvram_model_memory = 0.1 loaded_model.model_load(lowvram_model_memory, force_patch_weights=force_patch_weights) From 489fc75791cf5bc9628224ad1df06e494ed3f954 Mon Sep 17 00:00:00 2001 From: fromfirstbyte Date: Thu, 25 Jun 2026 09:00:24 +0300 Subject: [PATCH 2/2] replaced vram_state = VRAMState.SHARED condition with is_device_mps() --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 374f5503d..952cdc3fd 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -938,7 +938,7 @@ def load_models_gpu(models, memory_required=0, force_patch_weights=False, minimu if lowvram_model_memory == 0: lowvram_model_memory = 0.1 - if vram_set_state == VRAMState.NO_VRAM or (set_vram_to == VRAMState.LOW_VRAM and vram_state == VRAMState.SHARED and not force_full_load): + if vram_set_state == VRAMState.NO_VRAM or (set_vram_to == VRAMState.LOW_VRAM and is_device_mps(torch_dev) and not force_full_load): lowvram_model_memory = 0.1 loaded_model.model_load(lowvram_model_memory, force_patch_weights=force_patch_weights)