Rename to `falcon_mamba`
Browse files- config.json +2 -2
config.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
"architectures": [
|
3 |
-
"
|
4 |
],
|
5 |
"conv_kernel": 4,
|
6 |
"eos_token_id": 11,
|
@@ -10,7 +10,7 @@
|
|
10 |
"initializer_range": 0.1,
|
11 |
"intermediate_size": 8192,
|
12 |
"layer_norm_epsilon": 1e-05,
|
13 |
-
"model_type": "
|
14 |
"num_hidden_layers": 64,
|
15 |
"pad_token_id": 0,
|
16 |
"rescale_prenorm_residual": false,
|
|
|
1 |
{
|
2 |
"architectures": [
|
3 |
+
"FalconMambaForCausalLM"
|
4 |
],
|
5 |
"conv_kernel": 4,
|
6 |
"eos_token_id": 11,
|
|
|
10 |
"initializer_range": 0.1,
|
11 |
"intermediate_size": 8192,
|
12 |
"layer_norm_epsilon": 1e-05,
|
13 |
+
"model_type": "falcon_mamba",
|
14 |
"num_hidden_layers": 64,
|
15 |
"pad_token_id": 0,
|
16 |
"rescale_prenorm_residual": false,
|