Fix potential lowvram issue.

6 months ago · 83f343146a
parent b021cf67c7
commit 83f343146a
1 changed files with 1 additions and 1 deletions
--- 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)