From 1368c20d885afa2bf3776476a173b9ae9e0737b9 Mon Sep 17 00:00:00 2001 From: aayushbaluni <73417844+aayushbaluni@users.noreply.github.com> Date: Tue, 17 Mar 2026 16:13:21 +0530 Subject: [PATCH] fix: continue instead of break for parametrizations entries Skip parametrizations.* entries and continue checking remaining params so mixed modules with both parametrized and non-parametrized children are correctly classified. --- comfy/model_patcher.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index 0d4471d7f..51a35fd75 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -736,8 +736,9 @@ 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: - if not name.startswith("parametrizations."): - default = True # default random weights in non leaf modules + if name.startswith("parametrizations."): + continue + 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():