whisper / mlc-chat-config.json
ruihanglai's picture
init
51f2d45
raw
history blame
2.11 kB
{
"version": "0.1.0",
"model_type": "whisper",
"quantization": "q0f16",
"model_config": {
"vocab_size": 51866,
"num_hidden_layers": 32,
"num_mel_bins": 128,
"encoder_layers": 32,
"encoder_attention_heads": 20,
"decoder_layers": 32,
"decoder_attention_heads": 20,
"decoder_ffn_dim": 5120,
"encoder_ffn_dim": 5120,
"decoder_start_token_id": 50258,
"d_model": 1280,
"max_source_positions": 1500,
"max_target_positions": 448,
"pad_token_id": 50256,
"bos_token_id": 50257,
"eos_token_id": 50257,
"begin_suppress_tokens": [
220,
50257
],
"head_dim": 64,
"max_batch_size": 8,
"suppress_tokens": [],
"forced_decoder_ids": [],
"context_window_size": 1500,
"prefill_chunk_size": 15000,
"tensor_parallel_shards": 1
},
"vocab_size": 51866,
"context_window_size": 1500,
"sliding_window_size": -1,
"prefill_chunk_size": 15000,
"attention_sink_size": -1,
"tensor_parallel_shards": 1,
"temperature": 1.0,
"presence_penalty": 0.0,
"frequency_penalty": 0.0,
"repetition_penalty": 1.0,
"top_p": 1.0,
"tokenizer_files": [
"tokenizer.json",
"vocab.json",
"merges.txt",
"added_tokens.json",
"tokenizer_config.json"
],
"tokenizer_info": {
"token_postproc_method": "byte_level",
"prepend_space_in_encode": false,
"strip_space_in_decode": false
},
"conv_template": {
"name": "LM",
"system_template": "{system_message}",
"system_message": "",
"system_prefix_token_ids": [
1
],
"add_role_after_system_message": true,
"roles": {
"user": "",
"assistant": ""
},
"role_templates": {
"user": "{user_message}",
"assistant": "{assistant_message}",
"tool": "{tool_message}"
},
"messages": [],
"seps": [
""
],
"role_content_sep": "",
"role_empty_sep": "",
"stop_str": [],
"stop_token_ids": [
2
],
"function_string": "",
"use_function_calling": false
},
"pad_token_id": 50257,
"bos_token_id": 50257,
"eos_token_id": 50257
}