xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-09 13:47:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix the bug 'Linear' object has no attribute 'weight_function'

Browse Source 
fix the bug 'Linear' object has no attribute 'weight_function'
This commit is contained in:
qiaoce 2025-02-21 09:13:19 +08:00 committed by GitHub
parent cc9d669c62
commit 38ed808658
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_patcher.py
View File

@ -97,7 +97,7 @@ def wipe_lowvram_weight(m):
m.comfy_cast_weights = m.prev_comfy_cast_weights m.comfy_cast_weights = m.prev_comfy_cast_weights
del m.prev_comfy_cast_weights del m.prev_comfy_cast_weights
if not hasattr(m, "weight_function"): if hasattr(m, "weight_function"):
m.weight_function = [] m.weight_function = []
if hasattr(m, "bias_function"): if hasattr(m, "bias_function"):
@ -781,6 +781,8 @@ class ModelPatcher:
if move_weight: if move_weight:
cast_weight = self.force_cast_weights cast_weight = self.force_cast_weights
m.to(device_to) m.to(device_to)
if not hasattr(m, "weight_function"):
m.weight_function=[]
module_mem += move_weight_functions(m, device_to) module_mem += move_weight_functions(m, device_to)
if lowvram_possible: if lowvram_possible:
if weight_key in self.patches: if weight_key in self.patches: