DeepMount00 commited on
Commit
ab3f960
1 Parent(s): 636b97c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -22
config.json CHANGED
@@ -1,25 +1,11 @@
1
 
2
  {
3
- "architectures": [
4
- "MambaForCausalLM"
5
- ],
6
- "auto_map": {
7
- "AutoConfig": "configuration_mamba.MambaConfig",
8
- "AutoModelForCausalLM": "modeling_mamba.MambaForCausalLM"
9
- },
10
- "bias": false,
11
- "conv_bias": true,
12
- "d_conv": 4,
13
- "d_inner": 1536,
14
- "d_model": 768,
15
- "d_state": 16,
16
- "dt_rank": 48,
17
- "expand": 2,
18
- "initializer_range": 0.02,
19
- "model_type": "mamba",
20
- "n_layer": 24,
21
- "pad_vocab_size_multiple": 8,
22
- "torch_dtype": "float32",
23
- "transformers_version": "4.35.2",
24
- "vocab_size": 50280
25
  }
 
1
 
2
  {
3
+ "d_model": 1536,
4
+ "n_layer": 48,
5
+ "vocab_size": 50277,
6
+ "ssm_cfg": {},
7
+ "rms_norm": true,
8
+ "residual_in_fp32": true,
9
+ "fused_add_norm": true,
10
+ "pad_vocab_size_multiple": 8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  }