mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-12 20:48:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Automatically enable lowvram mode if vram is less than 4GB.

Browse Source 
Use: --normalvram to disable it.
This commit is contained in:
comfyanonymous
2023-02-10 00:47:56 -05:00
parent e9d3ac2ba0
commit 7e1e193f39
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
comfy/model_management.py
View File

@@ -13,16 +13,22 @@ total_vram_available_mb = -1
import sys
set_vram_to = NORMAL_VRAM
try:
import torch
total_vram = torch.cuda.mem_get_info(torch.cuda.current_device())[1] / (1024 * 1024)
if total_vram <= 4096 and not "--normalvram" in sys.argv:
print("Trying to enable lowvram mode because your GPU seems to have 4GB or less. If you don't want this use: --normalvram")
set_vram_to = LOW_VRAM
except:
pass
if "--lowvram" in sys.argv:
set_vram_to = LOW_VRAM
if "--novram" in sys.argv:
set_vram_to = NO_VRAM
try:
import torch
total_vram = torch.cuda.mem_get_info(torch.cuda.current_device())[1] / (1024 * 1024)
except:
pass
if set_vram_to != NORMAL_VRAM:
try: