From 83f343146ae1e8ccaf21da5b012bf59c78b97179 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 16 Aug 2024 17:12:42 -0400
Subject: [PATCH] Fix potential lowvram issue.

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

diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py
index 1edbf24..54a64aa 100644
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@@ -367,7 +367,7 @@ class ModelPatcher:
                 if mem_counter + module_mem >= lowvram_model_memory:
                     lowvram_weight = True
                     lowvram_counter += 1
-                    if m.comfy_cast_weights:
+                    if hasattr(m, "prev_comfy_cast_weights"): #Already lowvramed
                         continue
 
             weight_key = "{}.weight".format(n)