Upload modeling_esm_plusplus.py with huggingface_hub
Browse files- modeling_esm_plusplus.py +2 -1
modeling_esm_plusplus.py
CHANGED
@@ -533,7 +533,8 @@ class PreTrainedESMplusplusModel(PreTrainedModel):
|
|
533 |
if module.padding_idx is not None:
|
534 |
module.weight.data[module.padding_idx].zero_()
|
535 |
elif isinstance(module, nn.LayerNorm):
|
536 |
-
module.bias
|
|
|
537 |
module.weight.data.fill_(1.0)
|
538 |
|
539 |
@classmethod
|
|
|
533 |
if module.padding_idx is not None:
|
534 |
module.weight.data[module.padding_idx].zero_()
|
535 |
elif isinstance(module, nn.LayerNorm):
|
536 |
+
if module.bias is not None:
|
537 |
+
module.bias.data.zero_()
|
538 |
module.weight.data.fill_(1.0)
|
539 |
|
540 |
@classmethod
|