298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 10:53:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make --cpu have priority over everything else.

Browse Source
This commit is contained in:
comfyanonymous 2023-03-13 21:30:01 -04:00
parent 0e836d525e
commit ee46bef03a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management.py
View File

@ -50,8 +50,6 @@ if "--use-pytorch-cross-attention" in sys.argv:
XFORMERS_IS_AVAILBLE = False XFORMERS_IS_AVAILBLE = False
if "--cpu" in sys.argv:
vram_state = CPU
if "--lowvram" in sys.argv: if "--lowvram" in sys.argv:
set_vram_to = LOW_VRAM set_vram_to = LOW_VRAM
if "--novram" in sys.argv: if "--novram" in sys.argv:
@ -73,6 +71,8 @@ if set_vram_to == LOW_VRAM or set_vram_to == NO_VRAM:
total_vram_available_mb = (total_vram - 1024) // 2 total_vram_available_mb = (total_vram - 1024) // 2
total_vram_available_mb = int(max(256, total_vram_available_mb)) total_vram_available_mb = int(max(256, total_vram_available_mb))
if "--cpu" in sys.argv:
vram_state = CPU
print("Set vram state to:", ["CPU", "NO VRAM", "LOW VRAM", "NORMAL VRAM", "HIGH VRAM"][vram_state]) print("Set vram state to:", ["CPU", "NO VRAM", "LOW VRAM", "NORMAL VRAM", "HIGH VRAM"][vram_state])