do not use assing=True for nn.LayerNorm

AUTOMATIC1111 · Sep 25, 2024 · 5d32084 · 5d32084
1 parent 7d56c4a
commit 5d32084
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/sd_disable_initialization.py b/modules/sd_disable_initialization.py
@@ -176,7 +176,7 @@ def __enter__(self):
         def load_from_state_dict(original, module, state_dict, prefix, *args, **kwargs):
             used_param_keys = []
 
-            if isinstance(module, (torch.nn.Linear, torch.nn.Conv2d, torch.nn.GroupNorm, torch.nn.LayerNorm)):
+            if isinstance(module, (torch.nn.Linear, torch.nn.Conv2d, torch.nn.GroupNorm,)):
                 # HACK add assign=True to local_metadata for some cases
                 args[0]['assign_to_params_buffers'] = True