Upload config.json

config.json

@@ -1,22 +1,28 @@
 {
-  "architectures": [
-    "CoNNModel"
-  ],
-  "attention_probs_dropout_prob": 0.0,
+  "pad_token_id": 3,
+  "transformers_version": "4.26.1",
+  "vocab_size": 4,
+  "hidden_size": 132,
+  "num_hidden_layers": 4,
+  "num_attention_heads": 1,
   "hidden_act": "relu",
+  "intermediate_size": 42,
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 132,
+  "attention_probs_dropout_prob": 0.0,
+  "max_position_embeddings": 41,
   "initializer_range": 0.02,
-  "intermediate_size": 42,
   "layer_norm": false,
   "layer_norm_eps": 0.02,
-  "max_position_embeddings": 41,
   "mlp_hidden_size": 1959,
-  "model_type": "conn",
-  "num_attention_heads": 1,
-  "num_hidden_layers": 4,
-  "pad_token_id": 3,
-  "torch_dtype": "float32",
-  "transformers_version": "4.26.1",
-  "vocab_size": 4
+  "input_encoding_map": {
+    "0": 0,
+    "1": 1,
+    "bos": 2,
+    "pad": 3
+  },
+  "output_encoding_map": {
+    "0": 0,
+    "1": 1
+  },
+  "model_type": "conn"
 }