RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 59ac04ce2d into c9602625e4

This commit is contained in:
Roger 2026-07-19 08:13:34 +08:00 committed by GitHub
parent c9602625e4 59ac04ce2d
commit f8cf55817d
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 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): def key_param_name_to_key(key, param):
if len(key) == 0: if len(key) == 0:
return param return param
@ -950,8 +959,15 @@ class ModelPatcher:
weight_key = "{}.weight".format(n) weight_key = "{}.weight".format(n)
bias_key = "{}.bias".format(n) bias_key = "{}.bias".format(n)
if not full_load and hasattr(m, "comfy_cast_weights"): force_lazy_lora_patch = False
if not lowvram_fits: 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 offload_buffer = potential_offload
lowvram_weight = True lowvram_weight = True
lowvram_counter += 1 lowvram_counter += 1