Update modeling_me2bert.py
Browse files- modeling_me2bert.py +1 -1
modeling_me2bert.py
CHANGED
|
@@ -149,7 +149,7 @@ class ME2BertModel(PreTrainedModel):
|
|
| 149 |
|
| 150 |
def forward(self,
|
| 151 |
input_ids,
|
| 152 |
-
attention_mask, return_dict=False):
|
| 153 |
|
| 154 |
_, pooler_output = self.gen_feature_embeddings(
|
| 155 |
input_ids, attention_mask)
|
|
|
|
| 149 |
|
| 150 |
def forward(self,
|
| 151 |
input_ids,
|
| 152 |
+
attention_mask, return_dict=False, **kwargs):
|
| 153 |
|
| 154 |
_, pooler_output = self.gen_feature_embeddings(
|
| 155 |
input_ids, attention_mask)
|