Skip to content

GitLab

Unverified Commit cf0098d5 authored by Francesco Yoshi Gobbo's avatar Francesco Yoshi Gobbo
Browse files

no lowvram state if cpu only

parent 48efadec
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
comfy/model_management.py
View file @ cf0098d5
...@@ -22,7 +22,8 @@ try: ...@@ -22,7 +22,8 @@ try:
total_vram = torch.cuda.mem_get_info(torch.cuda.current_device())[1] / (1024 * 1024) total_vram = torch.cuda.mem_get_info(torch.cuda.current_device())[1] / (1024 * 1024)
total_ram = psutil.virtual_memory().total / (1024 * 1024) total_ram = psutil.virtual_memory().total / (1024 * 1024)
forced_normal_vram = "--normalvram" in sys.argv forced_normal_vram = "--normalvram" in sys.argv
if not forced_normal_vram: forced_cpu = "--cpu" in sys.argv
if not forced_normal_vram and not forced_cpu:
if total_vram <= 4096: if total_vram <= 4096:
print("Trying to enable lowvram mode because your GPU seems to have 4GB or less. If you don't want this use: --normalvram") 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 set_vram_to = LOW_VRAM
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment