From ee46bef03a98903831c01d31094a0c30ea411b28 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Mon, 13 Mar 2023 21:30:01 -0400
Subject: [PATCH] Make --cpu have priority over everything else.

---
 comfy/model_management.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 482b1add..c26d682f 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -50,8 +50,6 @@ if "--use-pytorch-cross-attention" in sys.argv:
     XFORMERS_IS_AVAILBLE = False
 
 
-if "--cpu" in sys.argv:
-    vram_state = CPU
 if "--lowvram" in sys.argv:
     set_vram_to = LOW_VRAM
 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 = 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])