guymorganb commited on
Commit
6db9dbd
·
1 Parent(s): 7d0cd1a

updated BertModel.__init__ call

Browse files
Files changed (1) hide show
  1. modeling_lsg_bert.py +4 -1
modeling_lsg_bert.py CHANGED
@@ -1023,8 +1023,11 @@ class LSGBertModel(LSGBertPreTrainedModel, BertModel):
1023
 
1024
  def __init__(self, config, add_pooling_layer=True):
1025
 
 
 
 
1026
  LSGBertPreTrainedModel.__init__(self, config)
1027
-
1028
  self.config = config
1029
 
1030
  self.embeddings = LSGBertEmbeddings(config)
 
1023
 
1024
  def __init__(self, config, add_pooling_layer=True):
1025
 
1026
+ # ensure your LSGBertModel inherits all the necessary fields introduced in the latest Transformers.
1027
+ BertModel.__init__(self, config)
1028
+
1029
  LSGBertPreTrainedModel.__init__(self, config)
1030
+
1031
  self.config = config
1032
 
1033
  self.embeddings = LSGBertEmbeddings(config)