Spaces:
Running
on
L4
Running
on
L4
Update MagicQuill/comfy/model_management.py
Browse files
MagicQuill/comfy/model_management.py
CHANGED
|
@@ -20,8 +20,8 @@ class CPUState(Enum):
|
|
| 20 |
MPS = 2
|
| 21 |
|
| 22 |
# Determine VRAM State
|
| 23 |
-
vram_state = VRAMState.
|
| 24 |
-
set_vram_to = VRAMState.
|
| 25 |
cpu_state = CPUState.GPU
|
| 26 |
|
| 27 |
total_vram = 0
|
|
|
|
| 20 |
MPS = 2
|
| 21 |
|
| 22 |
# Determine VRAM State
|
| 23 |
+
vram_state = VRAMState.LOW_VRAM
|
| 24 |
+
set_vram_to = VRAMState.LOW_VRAM
|
| 25 |
cpu_state = CPUState.GPU
|
| 26 |
|
| 27 |
total_vram = 0
|