Fix potential lowvram issue.

This commit is contained in:
comfyanonymous 2024-08-16 17:12:42 -04:00
parent b021cf67c7
commit 83f343146a

View File

@ -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)