mlc-rwkv5 / mlc-chat-config.json
sageking's picture
Upload my pretrained model
a583101
{
"model_type": "rwkv5",
"quantization": "q4f16_1",
"model_config": {
"hidden_size": 4096,
"intermediate_size": 14336,
"num_hidden_layers": 32,
"vocab_size": 65536,
"model_version": "5_2",
"tensor_parallel_shards": 1,
"rescale_every": 6,
"head_size": 64,
"layer_norm_epsilon": 1e-05,
"context_window_size": -1,
"prefill_chunk_size": 4096,
"num_heads": 64,
"max_batch_size": 80
},
"vocab_size": 65536,
"context_window_size": -1,
"sliding_window_size": -1,
"prefill_chunk_size": 4096,
"attention_sink_size": -1,
"tensor_parallel_shards": 1,
"mean_gen_len": 128,
"max_gen_len": 512,
"shift_fill_factor": 0.3,
"temperature": 0.7,
"presence_penalty": 0.0,
"frequency_penalty": 0.0,
"repetition_penalty": 1.0,
"top_p": 0.1,
"conv_template": {
"name": "rwkv_world",
"system_template": "User: hi\n\nAssistant: {system_message}",
"system_message": "Hi. I am your assistant and I will provide expert full response in full details. Please feel free to ask any question and I will always answer it.",
"add_role_after_system_message": true,
"roles": {
"user": "User",
"assistant": "Assistant"
},
"role_templates": {
"user": "{user_message}",
"assistant": "{assistant_message}",
"tool": "{tool_message}"
},
"messages": [],
"seps": [
"\n\n"
],
"role_content_sep": ": ",
"role_empty_sep": ": ",
"stop_str": [
"\n\n"
],
"stop_token_ids": [
0
],
"function_string": "",
"use_function_calling": false
},
"pad_token_id": 0,
"bos_token_id": 0,
"eos_token_id": 0,
"tokenizer_files": [
"tokenizer_config.json"
],
"token_table_postproc_method": "byte_fallback",
"version": "0.1.0"
}