Tom Aarsen
commited on
Commit
·
dd59cdc
1
Parent(s):
c10259c
Update AutoModel... in config.json
Browse filesPreviously, loading an AutoModel.from_pretrained actually loaded the Masked language modeling model, and it was not possible to load the Sequence Classification model.
- config.json +3 -1
config.json
CHANGED
|
@@ -4,7 +4,9 @@
|
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "model.NeoBERTConfig",
|
| 7 |
-
"AutoModel": "model.
|
|
|
|
|
|
|
| 8 |
},
|
| 9 |
"classifier_init_range": 0.02,
|
| 10 |
"decoder_init_range": 0.02,
|
|
|
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "model.NeoBERTConfig",
|
| 7 |
+
"AutoModel": "model.NeoBERT",
|
| 8 |
+
"AutoModelForMaskedLM": "model.NeoBERTLMHead",
|
| 9 |
+
"AutoModelForSequenceClassification": "model.NeoBERTForSequenceClassification"
|
| 10 |
},
|
| 11 |
"classifier_init_range": 0.02,
|
| 12 |
"decoder_init_range": 0.02,
|