Upload folder using huggingface_hub
Browse files- config.json +6 -1
- configuration_pantagruel_uni.py +5 -0
config.json
CHANGED
|
@@ -8,7 +8,12 @@
|
|
| 8 |
"attention_dropout": 0.1,
|
| 9 |
"auto_map": {
|
| 10 |
"AutoConfig": "configuration_pantagruel_uni.PantagruelUniConfig",
|
| 11 |
-
"AutoModel": "modeling_pantagruel_uni.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
},
|
| 13 |
"bad_words_ids": null,
|
| 14 |
"begin_suppress_tokens": null,
|
|
|
|
| 8 |
"attention_dropout": 0.1,
|
| 9 |
"auto_map": {
|
| 10 |
"AutoConfig": "configuration_pantagruel_uni.PantagruelUniConfig",
|
| 11 |
+
"AutoModel": "modeling_pantagruel_uni.PantagruelUniModel",
|
| 12 |
+
"AutoModelForMaskedLM": "modeling_pantagruel_uni.PantagruelUniForMaskedLM",
|
| 13 |
+
"AutoModelForMultipleChoice": "modeling_pantagruel_uni.PantagruelUniForMultipleChoice",
|
| 14 |
+
"AutoModelForQuestionAnswering": "modeling_pantagruel_uni.PantagruelUniForQuestionAnswering",
|
| 15 |
+
"AutoModelForSequenceClassification": "modeling_pantagruel_uni.PantagruelUniForSequenceClassification",
|
| 16 |
+
"AutoModelForTokenClassification": "modeling_pantagruel_uni.PantagruelUniForTokenClassification"
|
| 17 |
},
|
| 18 |
"bad_words_ids": null,
|
| 19 |
"begin_suppress_tokens": null,
|
configuration_pantagruel_uni.py
CHANGED
|
@@ -415,4 +415,9 @@ class PantagruelUniConfig(MyPretrainedConfig):
|
|
| 415 |
self.auto_map = {
|
| 416 |
'AutoConfig': 'configuration_pantagruel_uni.PantagruelUniConfig',
|
| 417 |
'AutoModel': 'modeling_pantagruel_uni.PantagruelUniModel',
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 418 |
}
|
|
|
|
| 415 |
self.auto_map = {
|
| 416 |
'AutoConfig': 'configuration_pantagruel_uni.PantagruelUniConfig',
|
| 417 |
'AutoModel': 'modeling_pantagruel_uni.PantagruelUniModel',
|
| 418 |
+
'AutoModelForMaskedLM': 'modeling_pantagruel_uni.PantagruelUniForMaskedLM',
|
| 419 |
+
'AutoModelForSequenceClassification': 'modeling_pantagruel_uni.PantagruelUniForSequenceClassification',
|
| 420 |
+
'AutoModelForMultipleChoice': 'modeling_pantagruel_uni.PantagruelUniForMultipleChoice',
|
| 421 |
+
'AutoModelForTokenClassification': 'modeling_pantagruel_uni.PantagruelUniForTokenClassification',
|
| 422 |
+
'AutoModelForQuestionAnswering': 'modeling_pantagruel_uni.PantagruelUniForQuestionAnswering',
|
| 423 |
}
|