hf-transformers-bot commited on
Commit
c75dcdf
·
verified ·
1 Parent(s): 9d559d3

Update tiny models for DbrxModel

Browse files
Files changed (4) hide show
  1. config.json +47 -0
  2. model.safetensors +3 -0
  3. tokenizer.json +0 -0
  4. tokenizer_config.json +13 -0
config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "DbrxModel"
4
+ ],
5
+ "attn_config": {
6
+ "attn_pdrop": 0.0,
7
+ "clip_qkv": 8,
8
+ "kv_n_heads": 1,
9
+ "model_type": "",
10
+ "rope_theta": 500000
11
+ },
12
+ "bos_token_id": 50256,
13
+ "d_model": 32,
14
+ "dtype": "float32",
15
+ "emb_pdrop": 0.0,
16
+ "eos_token_id": 50256,
17
+ "ffn_config": {
18
+ "ffn_act_fn": {
19
+ "name": "gelu"
20
+ },
21
+ "ffn_hidden_size": 64,
22
+ "hidden_size": 32,
23
+ "model_type": "",
24
+ "moe_jitter_eps": 0.0,
25
+ "moe_loss_weight": 0.05,
26
+ "moe_normalize_expert_weights": 1.0,
27
+ "moe_num_experts": 8,
28
+ "moe_top_k": 4
29
+ },
30
+ "initializer_range": 0.02,
31
+ "max_seq_len": 512,
32
+ "model_type": "dbrx",
33
+ "n_heads": 2,
34
+ "n_layers": 2,
35
+ "num_key_value_heads": 1,
36
+ "output_router_logits": false,
37
+ "pad_token_id": 0,
38
+ "resid_pdrop": 0.0,
39
+ "rope_parameters": {
40
+ "rope_theta": 10000.0,
41
+ "rope_type": "default"
42
+ },
43
+ "tie_word_embeddings": false,
44
+ "transformers_version": "5.16.0.dev0",
45
+ "use_cache": true,
46
+ "vocab_size": 50257
47
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5b024922d9f29630c7d6c8cb3118455bf30f5c2cde37c5eef11b1d268821312
3
+ size 6855160
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|endoftext|>",
5
+ "eos_token": "<|endoftext|>",
6
+ "errors": "replace",
7
+ "is_local": true,
8
+ "local_files_only": false,
9
+ "model_max_length": 512,
10
+ "pad_token": null,
11
+ "tokenizer_class": "GPT2Tokenizer",
12
+ "unk_token": "<|endoftext|>"
13
+ }