Commit
·
3154d93
1
Parent(s):
5b03f30
feat: set proper model_type
Browse files
configuration_xlm_roberta.py
CHANGED
|
@@ -5,6 +5,9 @@ from transformers import PretrainedConfig
|
|
| 5 |
|
| 6 |
|
| 7 |
class XLMRobertaFlashConfig(PretrainedConfig):
|
|
|
|
|
|
|
|
|
|
| 8 |
def __init__(
|
| 9 |
self,
|
| 10 |
vocab_size: int = 250002,
|
|
|
|
| 5 |
|
| 6 |
|
| 7 |
class XLMRobertaFlashConfig(PretrainedConfig):
|
| 8 |
+
|
| 9 |
+
model_type = "xlm-roberta"
|
| 10 |
+
|
| 11 |
def __init__(
|
| 12 |
self,
|
| 13 |
vocab_size: int = 250002,
|