wheeler404 commited on
Commit
bc5aa37
·
verified ·
1 Parent(s): 9d71641

Add Qwen2.5-tiny model with single hidden layer

Browse files
Files changed (3) hide show
  1. config.json +9 -3
  2. generation_config.json +1 -1
  3. model.safetensors +2 -2
config.json CHANGED
@@ -9,19 +9,25 @@
9
  "hidden_size": 64,
10
  "initializer_range": 0.02,
11
  "intermediate_size": 4864,
 
 
 
 
 
 
12
  "max_position_embeddings": 327,
13
  "max_window_layers": 24,
14
  "model_type": "qwen2",
15
  "num_attention_heads": 6,
16
- "num_hidden_layers": 1,
17
  "num_key_value_heads": 2,
18
  "rms_norm_eps": 1e-06,
19
  "rope_scaling": null,
20
  "rope_theta": 1000000.0,
21
- "sliding_window": 327,
22
  "tie_word_embeddings": true,
23
  "torch_dtype": "bfloat16",
24
- "transformers_version": "4.52.4",
25
  "use_cache": true,
26
  "use_mrope": false,
27
  "use_sliding_window": false,
 
9
  "hidden_size": 64,
10
  "initializer_range": 0.02,
11
  "intermediate_size": 4864,
12
+ "layer_types": [
13
+ "full_attention",
14
+ "full_attention",
15
+ "full_attention",
16
+ "full_attention"
17
+ ],
18
  "max_position_embeddings": 327,
19
  "max_window_layers": 24,
20
  "model_type": "qwen2",
21
  "num_attention_heads": 6,
22
+ "num_hidden_layers": 4,
23
  "num_key_value_heads": 2,
24
  "rms_norm_eps": 1e-06,
25
  "rope_scaling": null,
26
  "rope_theta": 1000000.0,
27
+ "sliding_window": null,
28
  "tie_word_embeddings": true,
29
  "torch_dtype": "bfloat16",
30
+ "transformers_version": "4.55.0",
31
  "use_cache": true,
32
  "use_mrope": false,
33
  "use_sliding_window": false,
generation_config.json CHANGED
@@ -2,5 +2,5 @@
2
  "_from_model_config": true,
3
  "bos_token_id": 151643,
4
  "eos_token_id": 151643,
5
- "transformers_version": "4.52.4"
6
  }
 
2
  "_from_model_config": true,
3
  "bos_token_id": 151643,
4
  "eos_token_id": 151643,
5
+ "transformers_version": "4.55.0"
6
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:37039fe993775b47e3ab5ce55c09a9408a638989a0497db3af5b11755d3c21ed
3
- size 21338160
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78c041d8be4c4764f840bab7d8d5ff2a09668f7194599508494bcb0ffd59fd6a
3
+ size 27008160