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)