Update modeling_gptbert.py
Browse files- modeling_gptbert.py +0 -4
modeling_gptbert.py
CHANGED
|
@@ -983,7 +983,6 @@ class GptBertForCausalLM(GptBertModel):
|
|
| 983 |
|
| 984 |
class GptBertForSequenceClassification(GptBertModel):
|
| 985 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 986 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 987 |
|
| 988 |
def __init__(self, config, **kwargs):
|
| 989 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -1049,7 +1048,6 @@ class GptBertForSequenceClassification(GptBertModel):
|
|
| 1049 |
|
| 1050 |
class GptBertForTokenClassification(GptBertModel):
|
| 1051 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 1052 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 1053 |
|
| 1054 |
def __init__(self, config, **kwargs):
|
| 1055 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -1097,7 +1095,6 @@ class GptBertForTokenClassification(GptBertModel):
|
|
| 1097 |
|
| 1098 |
class GptBertForQuestionAnswering(GptBertModel):
|
| 1099 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 1100 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 1101 |
|
| 1102 |
def __init__(self, config, **kwargs):
|
| 1103 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -1165,7 +1162,6 @@ class GptBertForQuestionAnswering(GptBertModel):
|
|
| 1165 |
|
| 1166 |
class GptBertForMultipleChoice(GptBertModel):
|
| 1167 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 1168 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 1169 |
|
| 1170 |
def __init__(self, config, **kwargs):
|
| 1171 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 983 |
|
| 984 |
class GptBertForSequenceClassification(GptBertModel):
|
| 985 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 986 |
|
| 987 |
def __init__(self, config, **kwargs):
|
| 988 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 1048 |
|
| 1049 |
class GptBertForTokenClassification(GptBertModel):
|
| 1050 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 1051 |
|
| 1052 |
def __init__(self, config, **kwargs):
|
| 1053 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 1095 |
|
| 1096 |
class GptBertForQuestionAnswering(GptBertModel):
|
| 1097 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 1098 |
|
| 1099 |
def __init__(self, config, **kwargs):
|
| 1100 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 1162 |
|
| 1163 |
class GptBertForMultipleChoice(GptBertModel):
|
| 1164 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 1165 |
|
| 1166 |
def __init__(self, config, **kwargs):
|
| 1167 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|