wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-18 12:28:17 +08:00
Code Issues Actions 26 Packages Projects Releases Wiki Activity

Merge 59ac04ce2d into 8edea4a65d

Browse Source
This commit is contained in:
Roger 2026-07-18 00:25:30 +08:00 committed by GitHub
commit 88c0acf4f3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
comfy/model_patcher.py
View File

@ -189,6 +189,15 @@ def get_key_weight(model, key):
return weight, set_func, convert_func
def is_quantized_lora_target(model, key):
try:
weight, set_func, convert_func = get_key_weight(model, key)
except Exception:
return False
return isinstance(weight, QuantizedTensor)
def key_param_name_to_key(key, param):
if len(key) == 0:
return param
@ -950,8 +959,15 @@ class ModelPatcher:
weight_key = "{}.weight".format(n)
bias_key = "{}.bias".format(n)
if not full_load and hasattr(m, "comfy_cast_weights"):
if not lowvram_fits:
force_lazy_lora_patch = False
if not force_patch_weights:
if weight_key in self.patches and is_quantized_lora_target(self.model, weight_key):
force_lazy_lora_patch = True
if bias_key in self.patches and is_quantized_lora_target(self.model, bias_key):
force_lazy_lora_patch = True
if hasattr(m, "comfy_cast_weights"):
if force_lazy_lora_patch or (not full_load and not lowvram_fits):
offload_buffer = potential_offload
lowvram_weight = True
lowvram_counter += 1