TRI-ML
/

mamba-7b-rw

Text Generation

Model card Files Files and versions Community

sedrickkeh commited on Apr 16

Commit

4e33340

•

1 Parent(s): e8d8450

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -25,7 +25,7 @@
   "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": "float32",
   "transformers_version": "4.39.0.dev0",

   "time_step_init_scheme": "random",
   "time_step_max": 0.1,
   "time_step_min": 0.001,
+  "time_step_rank": 256,
   "time_step_scale": 1.0,
   "torch_dtype": "float32",
   "transformers_version": "4.39.0.dev0",