Skip to content

Commit

Permalink
Fix potential lowvram issue.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Aug 16, 2024
1 parent b021cf6 commit 83f3431
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion comfy/model_patcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -367,7 +367,7 @@ def lowvram_load(self, device_to=None, lowvram_model_memory=0, force_patch_weigh
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)
Expand Down

0 comments on commit 83f3431

Please sign in to comment.