Spaces:
Running
Running
--- a/fastchat/conversation.py | |
+++ b/fastchat/conversation.py | |
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|>", | |
) | |
) | |