tiiuae
/

falcon-mamba-7b

Text Generation

Inference Endpoints

Model card Files Files and versions Community

ybelkada commited on Jul 23

Commit

5337fd7

•

1 Parent(s): 470ed3c

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "./",
   "architectures": [
-    "MambaForCausalLM"
   ],
   "bos_token_id": 0,
   "conv_kernel": 4,
@@ -12,7 +12,7 @@
   "initializer_range": 0.1,
   "intermediate_size": 8192,
   "layer_norm_epsilon": 1e-05,
-  "model_type": "mamba",
   "num_hidden_layers": 64,
   "pad_token_id": 11,
   "rescale_prenorm_residual": false,

 {
   "_name_or_path": "./",
   "architectures": [
+    "FalconMambaForCausalLM"
   ],
   "bos_token_id": 0,
   "conv_kernel": 4,
   "initializer_range": 0.1,
   "intermediate_size": 8192,
   "layer_norm_epsilon": 1e-05,
+  "model_type": "falcon_mamba",
   "num_hidden_layers": 64,
   "pad_token_id": 11,
   "rescale_prenorm_residual": false,