Text Generation
Transformers
Safetensors
English
doge
pt
conversational
JingzeShi commited on
Commit
f62f006
·
verified ·
1 Parent(s): aaae83e

Upload DogeForCausalLM

Browse files
Files changed (3) hide show
  1. config.json +11 -20
  2. generation_config.json +4 -7
  3. model.safetensors +2 -2
config.json CHANGED
@@ -1,43 +1,34 @@
1
  {
2
- "_name_or_path": "SmallDoge/Doge-20M",
3
  "architectures": [
4
  "DogeForCausalLM"
5
  ],
 
6
  "attention_dropout": 0.0,
7
- "auto_map": {
8
- "AutoConfig": "configuration_doge.DogeConfig",
9
- "AutoModelForCausalLM": "modeling_doge.DogeForCausalLM"
10
- },
11
- "bos_token_id": 0,
12
- "dynamic_mask_ratio": 0.0,
13
- "eos_token_id": 1,
14
- "expert_retrieval_size": 256,
15
  "hidden_act": "silu",
16
- "hidden_bias": false,
17
  "hidden_dropout": 0.0,
18
  "hidden_size": 256,
19
  "initializer_range": 0.02,
20
  "intermediate_size": 512,
21
  "is_moe": false,
 
22
  "max_position_embeddings": 2048,
 
23
  "model_type": "doge",
 
24
  "num_attention_heads": 2,
25
- "num_cdmoe_experts": 16348,
26
- "num_cdmoe_experts_per_head": 8,
27
- "num_cdmoe_heads": 4,
28
  "num_hidden_layers": 8,
29
  "num_key_value_heads": 1,
30
- "pad_token_id": 2,
31
  "rms_norm_eps": 1e-06,
32
- "rope_scaling": {
33
- "factor": 4.0,
34
- "original_max_position_embeddings": 2048,
35
- "rope_type": "dynamic"
36
- },
37
  "rope_theta": 10000.0,
 
 
38
  "tie_word_embeddings": true,
39
  "torch_dtype": "float32",
40
- "transformers_version": "4.48.3",
41
  "use_cache": true,
42
  "vocab_size": 32768
43
  }
 
1
  {
 
2
  "architectures": [
3
  "DogeForCausalLM"
4
  ],
5
+ "attention_bias": false,
6
  "attention_dropout": 0.0,
 
 
 
 
 
 
 
 
7
  "hidden_act": "silu",
 
8
  "hidden_dropout": 0.0,
9
  "hidden_size": 256,
10
  "initializer_range": 0.02,
11
  "intermediate_size": 512,
12
  "is_moe": false,
13
+ "keep_window_size": 2048,
14
  "max_position_embeddings": 2048,
15
+ "mlp_bias": false,
16
  "model_type": "doge",
17
+ "norm_topk_prob": false,
18
  "num_attention_heads": 2,
19
+ "num_experts": 16384,
20
+ "num_experts_per_tok": 64,
 
21
  "num_hidden_layers": 8,
22
  "num_key_value_heads": 1,
23
+ "output_router_logits": false,
24
  "rms_norm_eps": 1e-06,
25
+ "rope_scaling": null,
 
 
 
 
26
  "rope_theta": 10000.0,
27
+ "router_aux_loss_coef": 0.001,
28
+ "sliding_window": null,
29
  "tie_word_embeddings": true,
30
  "torch_dtype": "float32",
31
+ "transformers_version": "4.53.0.dev0",
32
  "use_cache": true,
33
  "vocab_size": 32768
34
  }
generation_config.json CHANGED
@@ -1,7 +1,4 @@
1
- {
2
- "_from_model_config": true,
3
- "bos_token_id": 0,
4
- "eos_token_id": 1,
5
- "pad_token_id": 2,
6
- "transformers_version": "4.48.3"
7
- }
 
1
+ {
2
+ "_from_model_config": true,
3
+ "transformers_version": "4.53.0.dev0"
4
+ }
 
 
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3632a5c94bc7d3cf66602318b168603ec19f1025e0aef01c286d65e30ed55e8b
3
- size 52482152
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac54b1befcb0193baa997e2434b58498a5a2f4c3f282b33e82ba748cb8030a8c
3
+ size 52487888