Upload TypeBERTForSequenceClassification
Browse files- type_bert_model.py +2 -0
type_bert_model.py
CHANGED
|
@@ -26,6 +26,8 @@ class TypeBERTConfig(PretrainedConfig):
|
|
| 26 |
"virtual": 4,
|
| 27 |
"concept": 5
|
| 28 |
}
|
|
|
|
|
|
|
| 29 |
|
| 30 |
|
| 31 |
|
|
|
|
| 26 |
"virtual": 4,
|
| 27 |
"concept": 5
|
| 28 |
}
|
| 29 |
+
|
| 30 |
+
self.architectures = ['TypeBERTForSequenceClassification']
|
| 31 |
|
| 32 |
|
| 33 |
|