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
|