Commit
·
6db9dbd
1
Parent(s):
7d0cd1a
updated BertModel.__init__ call
Browse files- 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)
|