jykim310 commited on
Commit
3369248
1 Parent(s): adb530c

Update mlc-chat-config.json

Browse files
Files changed (1) hide show
  1. mlc-chat-config.json +9 -8
mlc-chat-config.json CHANGED
@@ -31,16 +31,17 @@
31
  "repetition_penalty": 1.0,
32
  "top_p": 0.95,
33
  "conv_template": {
34
- "name": "llava",
35
- "system_template": "{system_message}",
36
- "system_message": "\n",
37
  "system_prefix_token_ids": [
38
  1
39
  ],
40
  "add_role_after_system_message": false,
41
  "roles": {
42
- "user": "USER",
43
- "assistant": "ASSISTANT"
 
44
  },
45
  "role_templates": {
46
  "user": "{user_message}",
@@ -51,10 +52,10 @@
51
  "seps": [
52
  " "
53
  ],
54
- "role_content_sep": ": ",
55
- "role_empty_sep": ":",
56
  "stop_str": [
57
- "</s>"
58
  ],
59
  "stop_token_ids": [
60
  2
 
31
  "repetition_penalty": 1.0,
32
  "top_p": 0.95,
33
  "conv_template": {
34
+ "name": "llama-2",
35
+ "system_template": "[INST] <<SYS>>\n{system_message}\n<</SYS>>\n\n",
36
+ "system_message": "You are a helpful, respectful and honest assistant.",
37
  "system_prefix_token_ids": [
38
  1
39
  ],
40
  "add_role_after_system_message": false,
41
  "roles": {
42
+ "user": "[INST]",
43
+ "assistant": "[/INST]",
44
+ "tool": "[INST]"
45
  },
46
  "role_templates": {
47
  "user": "{user_message}",
 
52
  "seps": [
53
  " "
54
  ],
55
+ "role_content_sep": " ",
56
+ "role_empty_sep": " ",
57
  "stop_str": [
58
+ "[INST]"
59
  ],
60
  "stop_token_ids": [
61
  2