lhallee commited on
Commit
ad076ec
·
verified ·
1 Parent(s): 8fd1c21

Upload modeling_esm_plusplus.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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.data.zero_()
 
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