omarmomen
/

sf_babylm_1

Model card Files Files and versions Community

omarmomen commited on Feb 3

Commit

9b430d5

•

1 Parent(s): 4ef095f

Update config.json

Files changed (1) hide show

config.json +4 -2

config.json CHANGED Viewed

@@ -1,10 +1,12 @@
 {
   "architectures": [
-    "StructFormerModel"
   ],
   "auto_map": {
     "AutoConfig": "structformer.StructFormerConfig",
-    "AutoModelForMaskedLM": "structformer.StructFormerModel"
   },
   "conv_size": 9,
   "dropatt": 0.1,

 {
   "architectures": [
+    "StructFormerModel",
+    "StructFormerModelForSequenceClassification"
   ],
   "auto_map": {
     "AutoConfig": "structformer.StructFormerConfig",
+    "AutoModelForMaskedLM": "structformer.StructFormerModel",
+    "AutoModelForSequenceClassification": "structformer.StructFormerModelForSequenceClassification"
   },
   "conv_size": 9,
   "dropatt": 0.1,