This commit is contained in:
Ayush Baluni 2026-04-02 17:33:11 +03:00 committed by GitHub
commit 536b12b658
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -733,6 +733,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:
if name.startswith("parametrizations."):
continue
default = True # default random weights in non leaf modules
break
if default and default_device is not None: