wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-30 13:33:42 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix: include weight_norm parametrizations in partial loading (#11855)

Browse Source 
Root cause: _load_list() treated parametrized (weight_norm) params missing
from recurse=False as default random weights, skipping GPU loading.
This commit is contained in:
aayushbaluni 2026-03-17 16:01:34 +05:30
parent 379fbd1a82
commit 78501b11c3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_patcher.py
View File

@ -736,7 +736,8 @@ class ModelPatcher:
params = { name: param for name, param in m.named_parameters(recurse=False) }
for name, param in m.named_parameters(recurse=True):
if name not in params:
default = True # default random weights in non leaf modules
if not name.startswith("parametrizations."):
default = True # default random weights in non leaf modules
break
if default and default_device is not None:
for param_name, param in params.items():