Nickinfinity commited on
Commit
3d9813c
1 Parent(s): cfcc8f9

Signed-off-by: Nickinfinity <nikhilg.me@gmail.com>

Files changed (1) hide show
  1. 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": "llama_default",
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,