vram_state = VRAMState.NO_VRAM set_vram_to = VRAMState.NO_VRAM

MagicQuill/comfy/model_management.py

@@ -20,8 +20,8 @@ class CPUState(Enum):
     MPS = 2
 
 # Determine VRAM State
-vram_state = VRAMState.NORMAL_VRAM
-set_vram_to = VRAMState.NORMAL_VRAM
+vram_state = VRAMState.NO_VRAM
+set_vram_to = VRAMState.NO_VRAM
 cpu_state = CPUState.GPU
 
 total_vram = 0
@@ -453,7 +453,7 @@ def load_models_gpu(models, memory_required=0, force_patch_weights=False):
         if vram_set_state == VRAMState.NO_VRAM:
             lowvram_model_memory = 64 * 1024 * 1024
 
-        cur_loaded_model = loaded_model.model_load(lowvram_model_memory, force_patch_weights=force_patch_weights)
+        cur_loaded_model = loaded_model.model_load(64 * 1024 * 1024, force_patch_weights=force_patch_weights)
         current_loaded_models.insert(0, loaded_model)
     return