Upload folder using huggingface_hub

config.json

@@ -6,12 +6,12 @@
   "bos_token_id": 0,
   "conv_kernel": 4,
   "eos_token_id": 0,
-  "expand": 2,
+  "expand": 4,
   "fused_add_norm": true,
   "hidden_act": "silu",
-  "hidden_size": 4,
+  "hidden_size": 8,
   "initializer_range": 0.1,
-  "intermediate_size": 6,
+  "intermediate_size": 32,
   "layer_norm_epsilon": 1e-05,
   "model_type": "mamba",
   "n_layer": 2,
@@ -21,17 +21,18 @@
   "rescale_prenorm_residual": false,
   "residual_in_fp32": true,
   "rms_norm": true,
-  "state_size": 16,
+  "state_size": 8,
   "time_step_floor": 0.0001,
   "time_step_init_scheme": "random",
   "time_step_max": 0.1,
   "time_step_min": 0.001,
   "time_step_rank": 160,
   "time_step_scale": 1.0,
-  "torch_dtype": "float16",
-  "transformers_version": "4.39.2",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.45.2",
   "use_bias": false,
   "use_cache": true,
   "use_conv_bias": true,
+  "use_mambapy": false,
   "vocab_size": 50280
 }

generation_config.json

@@ -3,5 +3,5 @@
   "bos_token_id": 0,
   "eos_token_id": 0,
   "pad_token_id": 0,
-  "transformers_version": "4.39.2"
+  "transformers_version": "4.45.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:08b7c8f8088e61205f8f12b807186900608972bad5daa99a99623095580ecc99
-size 413736
+oid sha256:dda3cbb122b916339e098328b1f09506d276e2be8d65f01a3555c44e40ab5a13
+size 855760

tokenizer_config.json

@@ -1,4 +1,6 @@
 {
+  "add_bos_token": false,
+  "add_eos_token": false,
   "add_prefix_space": false,
   "added_tokens_decoder": {
     "0": {