Hugging Face's logo

Spaces:
AI4Editing
/
MagicQuill
Running on L4

App Files Community
70

Update MagicQuill/comfy/model_management.py

#54
by Dragonstar666 - opened
base: refs/heads/main
←
from: refs/pr/54
Discussion Files changed
+2
-2
Files changed (1) hide show
  1. MagicQuill/comfy/model_management.py +2 -2
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.NORMAL_VRAM
24
- set_vram_to = VRAMState.NORMAL_VRAM
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