Update modeling_gptbert.py
Browse files- modeling_gptbert.py +0 -4
modeling_gptbert.py
CHANGED
|
@@ -977,7 +977,6 @@ class GptBertForCausalLM(GptBertModel):
|
|
| 977 |
|
| 978 |
class GptBertForSequenceClassification(GptBertModel):
|
| 979 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 980 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 981 |
|
| 982 |
def __init__(self, config, **kwargs):
|
| 983 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -1043,7 +1042,6 @@ class GptBertForSequenceClassification(GptBertModel):
|
|
| 1043 |
|
| 1044 |
class GptBertForTokenClassification(GptBertModel):
|
| 1045 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 1046 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 1047 |
|
| 1048 |
def __init__(self, config, **kwargs):
|
| 1049 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -1091,7 +1089,6 @@ class GptBertForTokenClassification(GptBertModel):
|
|
| 1091 |
|
| 1092 |
class GptBertForQuestionAnswering(GptBertModel):
|
| 1093 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 1094 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 1095 |
|
| 1096 |
def __init__(self, config, **kwargs):
|
| 1097 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -1159,7 +1156,6 @@ class GptBertForQuestionAnswering(GptBertModel):
|
|
| 1159 |
|
| 1160 |
class GptBertForMultipleChoice(GptBertModel):
|
| 1161 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 1162 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 1163 |
|
| 1164 |
def __init__(self, config, **kwargs):
|
| 1165 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 977 |
|
| 978 |
class GptBertForSequenceClassification(GptBertModel):
|
| 979 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 980 |
|
| 981 |
def __init__(self, config, **kwargs):
|
| 982 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 1042 |
|
| 1043 |
class GptBertForTokenClassification(GptBertModel):
|
| 1044 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 1045 |
|
| 1046 |
def __init__(self, config, **kwargs):
|
| 1047 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 1089 |
|
| 1090 |
class GptBertForQuestionAnswering(GptBertModel):
|
| 1091 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 1092 |
|
| 1093 |
def __init__(self, config, **kwargs):
|
| 1094 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 1156 |
|
| 1157 |
class GptBertForMultipleChoice(GptBertModel):
|
| 1158 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 1159 |
|
| 1160 |
def __init__(self, config, **kwargs):
|
| 1161 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|