rubentium commited on
Commit
34b4e9a
·
verified ·
1 Parent(s): f7a9c97

Upload paper checkpoint l8k365m_dense_ropectl_a1p0_seed46 checkpoint-44000

Browse files
checkpoints/control_rope_365m/l8k365m_dense_ropectl_a1p0_seed46/checkpoint-44000/config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "vocab_size": 151936,
3
+ "hidden_size": 768,
4
+ "intermediate_size": 3072,
5
+ "num_hidden_layers": 28,
6
+ "num_attention_heads": 6,
7
+ "num_key_value_heads": 2,
8
+ "head_dim": 128,
9
+ "hidden_act": "silu",
10
+ "max_position_embeddings": 32768,
11
+ "initializer_range": 0.02,
12
+ "rms_norm_eps": 1e-06,
13
+ "use_cache": true,
14
+ "tie_word_embeddings": true,
15
+ "rope_theta": 1000000,
16
+ "rope_scaling": null,
17
+ "rope_train_position_scale_set": [
18
+ 1.0,
19
+ 0.9896339488176222,
20
+ 0.9692159692159692,
21
+ 0.9420289855072463,
22
+ 0.8730158730158729
23
+ ],
24
+ "attention_bias": false,
25
+ "attention_dropout": 0.0,
26
+ "bos_token_id": 151643,
27
+ "eos_token_id": 151643,
28
+ "checkpoint_activations": false,
29
+ "attention_type": "sparse_stochastic",
30
+ "sparse_block_size": 128,
31
+ "sparse_num_random_blocks": 31,
32
+ "sparse_mask_pool_size": 1,
33
+ "sparse_max_seq_length": 8192,
34
+ "sparse_layer_indices": null,
35
+ "sparse_mask_cache_dir": null,
36
+ "sparse_mask_seed": 0,
37
+ "sparse_mixed_nrb_set": null,
38
+ "moba_block_size": 128,
39
+ "moba_top_k": 32,
40
+ "asentmax_alpha": 1.5,
41
+ "asentmax_learn_temp": true,
42
+ "asentmax_init_temp": 1.0
43
+ }
checkpoints/control_rope_365m/l8k365m_dense_ropectl_a1p0_seed46/checkpoint-44000/loader_gen_state_rank0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:219fe840440dda68f0de970281c8b8624454a2ab1f92076e777ce874183f7083
3
+ size 6802
checkpoints/control_rope_365m/l8k365m_dense_ropectl_a1p0_seed46/checkpoint-44000/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e114e049bcadbad7df3635e18f052b9d6c6ca2486ff7faf92809bfdfd579c690
3
+ size 718031899
checkpoints/control_rope_365m/l8k365m_dense_ropectl_a1p0_seed46/checkpoint-44000/rng_state_rank0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b287241deeb9649dbc3481001fb882c0a9bd72fccc808c3452f8286d9631aef2
3
+ size 14821
checkpoints/control_rope_365m/l8k365m_dense_ropectl_a1p0_seed46/checkpoint-44000/training_state.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8b8f130701ef1da6a04f072f437b217080d8b82e03dc48bb1aab252006052036
3
+ size 1436105299