Sync latest GeneMamba docs and next-token training updates

config.json

@@ -1,16 +1,12 @@
 {
+  "model_type": "genemamba",
   "architectures": [
-    "MambaModel"
+    "GeneMambaModel"
   ],
-  "d_model": 512,
-  "mamba_layer": 24,
-  "torch_dtype": "float32",
-  "transformers_version": "4.40.2",
   "vocab_size": 25426,
+  "max_position_embeddings": 2048,
   "hidden_size": 512,
   "num_hidden_layers": 24,
-  "model_type": "genemamba",
-  "max_position_embeddings": 2048,
   "intermediate_size": 2048,
   "hidden_dropout_prob": 0.1,
   "initializer_range": 0.02,
@@ -18,13 +14,15 @@
   "embedding_pooling": "mean",
   "num_labels": 2,
   "pad_token_id": 1,
-  "bos_token_id": 0,
   "eos_token_id": 2,
+  "bos_token_id": 0,
   "use_cache": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.40.2",
   "auto_map": {
     "AutoConfig": "configuration_genemamba.GeneMambaConfig",
     "AutoModel": "modeling_genemamba.GeneMambaModel",
     "AutoModelForMaskedLM": "modeling_genemamba.GeneMambaForMaskedLM",
     "AutoModelForSequenceClassification": "modeling_genemamba.GeneMambaForSequenceClassification"
   }
-}
+}