wuyingwen/comfyui
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

Fix potential lowvram issue.

Browse Source
This commit is contained in:
comfyanonymous 2024-08-16 17:12:42 -04:00
parent b021cf67c7
commit 83f343146a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_patcher.py
View File

@ -367,7 +367,7 @@ class ModelPatcher:
if mem_counter + module_mem >= lowvram_model_memory: if mem_counter + module_mem >= lowvram_model_memory:
lowvram_weight = True lowvram_weight = True
lowvram_counter += 1 lowvram_counter += 1
if m.comfy_cast_weights: if hasattr(m, "prev_comfy_cast_weights"): #Already lowvramed
continue continue
weight_key = "{}.weight".format(n) weight_key = "{}.weight".format(n)