abdi38394 commited on
Commit
528b22e
·
verified ·
1 Parent(s): 639f52b

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.json +11 -2
  2. model.safetensors +3 -0
config.json CHANGED
@@ -2,12 +2,21 @@
2
  "architectures": [
3
  "MiniMoE50M"
4
  ],
 
5
  "vocab_size": 16000,
6
  "d_model": 512,
7
  "n_heads": 8,
8
  "d_ff": 1536,
9
  "n_layers": 5,
10
  "num_experts": 4,
11
- "routing_strategy": "top-1",
12
- "model_type": "custom-moe"
 
 
 
 
 
 
 
 
13
  }
 
2
  "architectures": [
3
  "MiniMoE50M"
4
  ],
5
+ "model_type": "mini-moe",
6
  "vocab_size": 16000,
7
  "d_model": 512,
8
  "n_heads": 8,
9
  "d_ff": 1536,
10
  "n_layers": 5,
11
  "num_experts": 4,
12
+ "task_specific_params": {
13
+ "text-generation": {
14
+ "do_sample": true,
15
+ "max_length": 50
16
+ }
17
+ },
18
+ "auto_map": {
19
+ "AutoConfig": "model.MiniMoEConfig",
20
+ "AutoModel": "model.MiniMoE50M"
21
+ }
22
  }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5acf901d2368ed8a41665521ca2a8dba516ae44b89b0c35c2537f644ca036e09
3
+ size 212640768