telecomadm1145 commited on
Commit
70e2e50
·
verified ·
1 Parent(s): 7ada0d9

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -9,13 +9,13 @@
9
  "expand": 2,
10
  "head_dim": 64,
11
  "hidden_act": "silu",
12
- "hidden_size": 512,
13
  "initializer_range": 0.1,
14
  "layer_norm_epsilon": 1e-05,
15
  "model_type": "mamba2",
16
  "n_groups": 1,
17
- "num_heads": 16,
18
- "num_hidden_layers": 36,
19
  "pad_token_id": 0,
20
  "rescale_prenorm_residual": false,
21
  "residual_in_fp32": true,
 
9
  "expand": 2,
10
  "head_dim": 64,
11
  "hidden_act": "silu",
12
+ "hidden_size": 1024,
13
  "initializer_range": 0.1,
14
  "layer_norm_epsilon": 1e-05,
15
  "model_type": "mamba2",
16
  "n_groups": 1,
17
+ "num_heads": 32,
18
+ "num_hidden_layers": 24,
19
  "pad_token_id": 0,
20
  "rescale_prenorm_residual": false,
21
  "residual_in_fp32": true,