Upload MambaForCausalLM

README.md

@@ -1,10 +1,10 @@
 ---
-library_name: transformers
+language:
+- en
 license: mit
+library_name: transformers
 datasets:
 - mlsquare/CLIENT_samantar_mixed_train_val
-language:
-- en
 pipeline_tag: text-generation
 ---
 

config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "mlsquare/pico_seshu_test",
   "architectures": [
     "MambaForCausalLM"
   ],
@@ -15,6 +16,6 @@
   "n_layer": 5,
   "pad_vocab_size_multiple": 8,
   "torch_dtype": "float32",
-  "transformers_version": "4.38.1",
+  "transformers_version": "4.38.2",
   "vocab_size": 7000
 }

generation_config.json

@@ -1,4 +1,4 @@
 {
   "_from_model_config": true,
-  "transformers_version": "4.38.1"
+  "transformers_version": "4.38.2"
 }