hf-transformers-bot commited on
Commit
db0a6b9
·
verified ·
1 Parent(s): b1ee7fe

Update tiny models for DeepseekV4ForCausalLM

Browse files
config.json ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "DeepseekV4ForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "compress_rates": {
9
+ "compressed_sparse_attention": 4,
10
+ "heavily_compressed_attention": 128
11
+ },
12
+ "compress_rope_theta": 160000.0,
13
+ "dtype": "float32",
14
+ "eos_token_id": 1,
15
+ "hc_eps": 1e-06,
16
+ "hc_mult": 2,
17
+ "hc_sinkhorn_iters": 3,
18
+ "head_dim": 32,
19
+ "hidden_act": "gelu",
20
+ "hidden_size": 64,
21
+ "index_head_dim": 16,
22
+ "index_n_heads": 2,
23
+ "index_topk": 2,
24
+ "initializer_range": 0.02,
25
+ "is_decoder": true,
26
+ "layer_types": [
27
+ "heavily_compressed_attention",
28
+ "compressed_sparse_attention"
29
+ ],
30
+ "max_position_embeddings": 200,
31
+ "mlp_bias": false,
32
+ "mlp_layer_types": [
33
+ "moe",
34
+ "moe"
35
+ ],
36
+ "model_type": "deepseek_v4",
37
+ "moe_intermediate_size": 64,
38
+ "n_routed_experts": 4,
39
+ "n_shared_experts": 1,
40
+ "norm_topk_prob": true,
41
+ "num_attention_heads": 4,
42
+ "num_experts_per_tok": 2,
43
+ "num_hidden_layers": 2,
44
+ "num_key_value_heads": 1,
45
+ "num_nextn_predict_layers": 0,
46
+ "o_groups": 2,
47
+ "o_lora_rank": 16,
48
+ "output_router_logits": false,
49
+ "pad_token_id": 1,
50
+ "partial_rotary_factor": 0.25,
51
+ "q_lora_rank": 32,
52
+ "qk_rope_head_dim": 8,
53
+ "rms_norm_eps": 1e-06,
54
+ "rope_parameters": {
55
+ "compress": {
56
+ "partial_rotary_factor": 0.25,
57
+ "rope_theta": 160000.0,
58
+ "rope_type": "default"
59
+ },
60
+ "main": {
61
+ "partial_rotary_factor": 0.25,
62
+ "rope_theta": 10000.0,
63
+ "rope_type": "default"
64
+ }
65
+ },
66
+ "rope_theta": 10000.0,
67
+ "routed_scaling_factor": 1.5,
68
+ "router_aux_loss_coef": 0.001,
69
+ "router_jitter_noise": 0.0,
70
+ "scoring_func": "sqrtsoftplus",
71
+ "sliding_window": 8,
72
+ "swiglu_limit": 10.0,
73
+ "tie_word_embeddings": false,
74
+ "transformers_version": "5.16.0.dev0",
75
+ "use_cache": true,
76
+ "vocab_size": 129280
77
+ }
generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 1,
5
+ "output_attentions": false,
6
+ "output_hidden_states": false,
7
+ "pad_token_id": 1,
8
+ "transformers_version": "5.16.0.dev0",
9
+ "use_cache": true
10
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8924be97b4f4f152420fbc5100f343c0e462378155acbcfc8c84e8c014b83bae
3
+ size 66899964
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "bos_token": "<|begin▁of▁sentence|>",
4
+ "clean_up_tokenization_spaces": false,
5
+ "eos_token": "<|end▁of▁sentence|>",
6
+ "is_local": true,
7
+ "legacy": true,
8
+ "local_files_only": false,
9
+ "model_max_length": 200,
10
+ "pad_token": "<|end▁of▁sentence|>",
11
+ "sp_model_kwargs": {},
12
+ "tokenizer_class": "TokenizersBackend",
13
+ "unk_token": null
14
+ }