From 89a84e32d2a771743ba10d6a6a0634f7e321fef5 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Mon, 26 May 2025 13:39:27 -0700
Subject: [PATCH] Disable initial GPU load when novram is used. (#8294)

---
 comfy/model_management.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 1e6e8553f..f5b37e68e 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -695,7 +695,7 @@ def unet_inital_load_device(parameters, dtype):
         return torch_dev
 
     cpu_dev = torch.device("cpu")
-    if DISABLE_SMART_MEMORY:
+    if DISABLE_SMART_MEMORY or vram_state == VRAMState.NO_VRAM:
         return cpu_dev
 
     model_size = dtype_size(dtype) * parameters