Nickinfinity
commited on
Commit
·
3d9813c
1
Parent(s):
cfcc8f9
update
Browse filesSigned-off-by: Nickinfinity <nikhilg.me@gmail.com>
- mlc-chat-config.json +5 -2
mlc-chat-config.json
CHANGED
@@ -13,7 +13,8 @@
|
|
13 |
"prefill_chunk_size": 768,
|
14 |
"num_key_value_heads": 4,
|
15 |
"head_dim": 64,
|
16 |
-
"tensor_parallel_shards": 1
|
|
|
17 |
},
|
18 |
"vocab_size": 32000,
|
19 |
"context_window_size": 768,
|
@@ -25,9 +26,11 @@
|
|
25 |
"max_gen_len": 512,
|
26 |
"shift_fill_factor": 0.3,
|
27 |
"temperature": 0.7,
|
|
|
|
|
28 |
"repetition_penalty": 1.0,
|
29 |
"top_p": 0.95,
|
30 |
-
"conv_template": "
|
31 |
"pad_token_id": 0,
|
32 |
"bos_token_id": 1,
|
33 |
"eos_token_id": 2,
|
|
|
13 |
"prefill_chunk_size": 768,
|
14 |
"num_key_value_heads": 4,
|
15 |
"head_dim": 64,
|
16 |
+
"tensor_parallel_shards": 1,
|
17 |
+
"max_batch_size": 80
|
18 |
},
|
19 |
"vocab_size": 32000,
|
20 |
"context_window_size": 768,
|
|
|
26 |
"max_gen_len": 512,
|
27 |
"shift_fill_factor": 0.3,
|
28 |
"temperature": 0.7,
|
29 |
+
"presence_penalty": 0.0,
|
30 |
+
"frequency_penalty": 0.0,
|
31 |
"repetition_penalty": 1.0,
|
32 |
"top_p": 0.95,
|
33 |
+
"conv_template": "llama-2",
|
34 |
"pad_token_id": 0,
|
35 |
"bos_token_id": 1,
|
36 |
"eos_token_id": 2,
|