Update modeling_hf_nomic_bert.py
Browse files
modeling_hf_nomic_bert.py
CHANGED
|
@@ -1915,7 +1915,6 @@ class NomicBertForMultipleChoice(NomicBertPreTrainedModel):
|
|
| 1915 |
attention_mask=attention_mask,
|
| 1916 |
token_type_ids=token_type_ids,
|
| 1917 |
position_ids=position_ids,
|
| 1918 |
-
head_mask=head_mask,
|
| 1919 |
inputs_embeds=inputs_embeds,
|
| 1920 |
)
|
| 1921 |
|
|
@@ -1980,7 +1979,6 @@ class NomicBertForTokenClassification(NomicBertPreTrainedModel):
|
|
| 1980 |
attention_mask=attention_mask,
|
| 1981 |
token_type_ids=token_type_ids,
|
| 1982 |
position_ids=position_ids,
|
| 1983 |
-
head_mask=head_mask,
|
| 1984 |
inputs_embeds=inputs_embeds,
|
| 1985 |
)
|
| 1986 |
|
|
@@ -2047,7 +2045,6 @@ class NomicBertForQuestionAnswering(NomicBertPreTrainedModel):
|
|
| 2047 |
attention_mask=attention_mask,
|
| 2048 |
token_type_ids=token_type_ids,
|
| 2049 |
position_ids=position_ids,
|
| 2050 |
-
head_mask=head_mask,
|
| 2051 |
inputs_embeds=inputs_embeds,
|
| 2052 |
)
|
| 2053 |
|
|
|
|
| 1915 |
attention_mask=attention_mask,
|
| 1916 |
token_type_ids=token_type_ids,
|
| 1917 |
position_ids=position_ids,
|
|
|
|
| 1918 |
inputs_embeds=inputs_embeds,
|
| 1919 |
)
|
| 1920 |
|
|
|
|
| 1979 |
attention_mask=attention_mask,
|
| 1980 |
token_type_ids=token_type_ids,
|
| 1981 |
position_ids=position_ids,
|
|
|
|
| 1982 |
inputs_embeds=inputs_embeds,
|
| 1983 |
)
|
| 1984 |
|
|
|
|
| 2045 |
attention_mask=attention_mask,
|
| 2046 |
token_type_ids=token_type_ids,
|
| 2047 |
position_ids=position_ids,
|
|
|
|
| 2048 |
inputs_embeds=inputs_embeds,
|
| 2049 |
)
|
| 2050 |
|