File size: 732 Bytes
a916f49
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
--- a/fastchat/conversation.py
+++ b/fastchat/conversation.py
@@ -787,12 +787,12 @@ def get_conv_template(name: str) -> Conversation:
     Conversation(
         name="xgen",
         system_message="A chat between a curious human and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the human's questions.\n\n",
-        roles=("### Human: ", "###"),
+        roles=("### Human", "### Assistant"),
         messages=(),
         offset=0,
-        sep_style=SeparatorStyle.NO_COLON_SINGLE,
+        sep_style=SeparatorStyle.ADD_COLON_SINGLE,
         sep="\n",
-        stop_token_ids=[50256, 0, 1, 2],
+        stop_token_ids=[50256],
         stop_str="<|endoftext|>",
     )
 )