OrionZheng commited on
Commit
5b8edbf
1 Parent(s): 96f1431

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +0 -51
config.json CHANGED
@@ -1,54 +1,3 @@
1
- {
2
- "architectures": [
3
- "OpenMoeForCausalLM"
4
- ],
5
- "auto_map": {
6
- "AutoModelForCausalLM": "modeling_openmoe.OpenMoeForCausalLM"
7
- },
8
- "attention_bias": false,
9
- "bos_token_id": 2,
10
- "dropout_rate": 0.0,
11
- "enable_comm_overlap": false,
12
- "enable_hierarchical_alltoall": false,
13
- "enable_kernel": false,
14
- "enable_load_balance": false,
15
- "eos_token_id": 1,
16
- "expert_parallel": null,
17
- "head_dim": 128,
18
- "hidden_act": "swiglu",
19
- "hidden_size": 2048,
20
- "initializer_range": 0.02,
21
- "intermediate_size": 8192,
22
- "layer_norm_epsilon": 1e-06,
23
- "load_balance_beam_width": 8,
24
- "load_balance_group_swap_factor": 0.4,
25
- "load_balance_tolerance": 0.1,
26
- "max_position_embeddings": 2048,
27
- "mlp_gated": true,
28
- "model_type": "llama",
29
- "moe_layer_interval": 6,
30
- "num_attention_heads": 24,
31
- "num_experts": 32,
32
- "num_hidden_layers": 24,
33
- "num_key_value_heads": 24,
34
- "pad_token_id": 0,
35
- "pretraining_tp": 1,
36
- "rms_norm_eps": 1e-06,
37
- "rope_scaling": null,
38
- "rope_theta": 10000.0,
39
- "router_capacity_factor_eval": 2.0,
40
- "router_capacity_factor_train": 1.25,
41
- "router_drop_tks": true,
42
- "router_min_capacity": 4,
43
- "router_noisy_policy": null,
44
- "router_topk": 2,
45
- "tie_word_embeddings": false,
46
- "torch_dtype": "float32",
47
- "transformers_version": "4.34.0",
48
- "use_cache": true,
49
- "vocab_size": 256384
50
- }
51
-
52
  {
53
  "architectures": [
54
  "OpenMoeForCausalLM"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  {
2
  "architectures": [
3
  "OpenMoeForCausalLM"