omarmomen
/

sf_ip_babylm_1

structformer_in_parser

Model card Files Files and versions Community

omarmomen commited on Feb 4

Commit

7286d0f

•

1 Parent(s): a1cab64

Update config.json

Files changed (1) hide show

config.json +4 -2

config.json CHANGED Viewed

@@ -1,10 +1,12 @@
 {
   "architectures": [
-    "StructFormer_In_ParserModel"
   ],
   "auto_map": {
     "AutoConfig": "structformer_in_parser.StructFormer_In_ParserConfig",
-    "AutoModelForMaskedLM": "structformer_in_parser.StructFormer_In_ParserModel"
   },
   "conv_size": 9,
   "dropatt": 0.1,

 {
   "architectures": [
+    "StructFormer_In_ParserModel",
+    "StructFormer_In_ParserModelForSequenceClassification"
   ],
   "auto_map": {
     "AutoConfig": "structformer_in_parser.StructFormer_In_ParserConfig",
+    "AutoModelForMaskedLM": "structformer_in_parser.StructFormer_In_ParserModel",
+    "AutoModelForSequenceClassification": "structformer_in_parser.StructFormer_In_ParserModelForSequenceClassification"
   },
   "conv_size": 9,
   "dropatt": 0.1,