RichardErkhov commited on
Commit
ae4da73
1 Parent(s): 4aa1dd6

uploaded model

Browse files
Files changed (1) hide show
  1. config.json +51 -0
config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "tiny-random-dbrx",
3
+ "architectures": [
4
+ "DbrxForCausalLM"
5
+ ],
6
+ "attn_config": {
7
+ "clip_qkv": 8,
8
+ "kv_n_heads": 2,
9
+ "model_type": "",
10
+ "rope_theta": 500000
11
+ },
12
+ "d_model": 8,
13
+ "emb_pdrop": 0.0,
14
+ "ffn_config": {
15
+ "ffn_hidden_size": 8,
16
+ "model_type": "",
17
+ "moe_jitter_eps": 0,
18
+ "moe_loss_weight": 0.05,
19
+ "moe_num_experts": 16,
20
+ "moe_top_k": 4
21
+ },
22
+ "initializer_range": 0.02,
23
+ "max_seq_len": 32768,
24
+ "model_type": "dbrx",
25
+ "n_heads": 4,
26
+ "n_layers": 2,
27
+ "num_key_value_heads": 2,
28
+ "output_router_logits": false,
29
+ "quantization_config": {
30
+ "_load_in_4bit": false,
31
+ "_load_in_8bit": true,
32
+ "bnb_4bit_compute_dtype": "float32",
33
+ "bnb_4bit_quant_storage": "uint8",
34
+ "bnb_4bit_quant_type": "fp4",
35
+ "bnb_4bit_use_double_quant": false,
36
+ "llm_int8_enable_fp32_cpu_offload": false,
37
+ "llm_int8_has_fp16_weight": false,
38
+ "llm_int8_skip_modules": null,
39
+ "llm_int8_threshold": 6.0,
40
+ "load_in_4bit": false,
41
+ "load_in_8bit": true,
42
+ "quant_method": "bitsandbytes"
43
+ },
44
+ "resid_pdrop": 0.0,
45
+ "router_aux_loss_coef": 0.05,
46
+ "tie_word_embeddings": false,
47
+ "torch_dtype": "float16",
48
+ "transformers_version": "4.44.2",
49
+ "use_cache": true,
50
+ "vocab_size": 100352
51
+ }