We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 75aa4dc commit c89c67cCopy full SHA for c89c67c
internlm/initialize/initialize_model.py
@@ -93,7 +93,7 @@ def set_fp32_attr_for_model(model: Union[nn.Module, nn.ModuleList]):
93
94
95
def set_parallel_attr_for_param_groups(model: Union[nn.Module, nn.ModuleList]):
96
- def _check_module(name, module):
+ def _check_module(name, module): # pylint: disable=W0613
97
# layer_norm
98
if isinstance(module, (RMSNorm, nn.LayerNorm)):
99
for param in module.parameters():
0 commit comments