omarmomen
/

tf_babylm_1

Model card Files Files and versions

omarmomen commited on Feb 3, 2024

Commit

5fdd4a3

·

verified ·

1 Parent(s): c799e10

Update config.json

Files changed (1) hide show

config.json +4 -2

config.json CHANGED Viewed

@@ -1,10 +1,12 @@
 {
   "architectures": [
-    "TransformerModel"
   ],
   "auto_map": {
     "AutoConfig": "structformer.TransformerConfig",
-    "AutoModelForMaskedLM": "structformer.TransformerModel"
   },
   "dropatt": 0.1,
   "dropout": 0.1,

 {
   "architectures": [
+    "TransformerModel",
+    "TransformerModelForSequenceClassification"
   ],
   "auto_map": {
     "AutoConfig": "structformer.TransformerConfig",
+    "AutoModelForMaskedLM": "structformer.TransformerModel",
+    "AutoModelForSequenceClassification": "structformer.TransformerModelForSequenceClassification"
   },
   "dropatt": 0.1,
   "dropout": 0.1,