From 78501b11c32611478ca9342ab06052794bc9a12d Mon Sep 17 00:00:00 2001 From: aayushbaluni <73417844+aayushbaluni@users.noreply.github.com> Date: Tue, 17 Mar 2026 16:01:34 +0530 Subject: [PATCH] fix: include weight_norm parametrizations in partial loading (#11855) Root cause: _load_list() treated parametrized (weight_norm) params missing from recurse=False as default random weights, skipping GPU loading. --- comfy/model_patcher.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index c26d37db2..0d4471d7f 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -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():