TinyLlama-1.1B-Chat-v0.4-q0f16-MLC / mlc-chat-config.json
ruihanglai's picture
Update mlc-chat-config.json
4a7238f verified
{
"model_type": "llama",
"quantization": "q0f16",
"model_config": {
"hidden_size": 2048,
"intermediate_size": 5632,
"num_attention_heads": 32,
"num_hidden_layers": 22,
"rms_norm_eps": 1e-05,
"vocab_size": 32003,
"position_embedding_base": 10000.0,
"context_window_size": 2048,
"prefill_chunk_size": 2048,
"num_key_value_heads": 4,
"head_dim": 64,
"tensor_parallel_shards": 1,
"max_batch_size": 80
},
"vocab_size": 32003,
"context_window_size": 2048,
"sliding_window_size": -1,
"prefill_chunk_size": 2048,
"attention_sink_size": -1,
"tensor_parallel_shards": 1,
"max_batch_size": 80,
"mean_gen_len": 128,
"max_gen_len": 512,
"shift_fill_factor": 0.3,
"temperature": 0.7,
"repetition_penalty": 1.0,
"top_p": 0.95,
"conv_template": {
"name": "chatml",
"system_template": "<|im_start|>system\n{system_message}",
"system_message": "A conversation between a user and an LLM-based AI assistant. The assistant gives helpful and honest answers.",
"add_role_after_system_message": true,
"roles": {
"user": "<|im_start|>user",
"assistant": "<|im_start|>assistant"
},
"role_templates": {
"user": "{user_message}",
"assistant": "{assistant_message}",
"tool": "{tool_message}"
},
"messages": [],
"seps": [
"<|im_end|>\n"
],
"role_content_sep": "\n",
"role_empty_sep": "\n",
"stop_str": [
"<|im_end|>"
],
"stop_token_ids": [
2
],
"function_string": "",
"use_function_calling": false
},
"pad_token_id": 0,
"bos_token_id": 1,
"eos_token_id": 2,
"tokenizer_files": [
"tokenizer.model",
"tokenizer.json",
"added_tokens.json",
"tokenizer_config.json"
],
"version": "0.1.0"
}