lianglv commited on
Commit
8073951
1 Parent(s): 5bcc7a6

Update conversation.py

Browse files
Files changed (1) hide show
  1. conversation.py +33 -0
conversation.py CHANGED
@@ -930,6 +930,39 @@ register_conv_template(
930
  )
931
  )
932
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
933
  # neuralchat-v2 template
934
  register_conv_template(
935
  Conversation(
 
930
  )
931
  )
932
 
933
+
934
+ # neuralchat-v3 template
935
+ register_conv_template(
936
+ Conversation(
937
+ name="neural-chat-7b-v3",
938
+ system_message="""### System:
939
+ - You are a helpful assistant chatbot trained by Intel.
940
+ - You answer questions.
941
+ - You are excited to be able to help the user, but will refuse to do anything that could be considered harmful to the user.
942
+ - You are more than just an information source, you are also able to write poetry, short stories, and make jokes.</s>\n""",
943
+ roles=("### User:", "### Assistant:"),
944
+ sep_style=SeparatorStyle.NO_COLON_TWO,
945
+ sep="\n",
946
+ sep2="</s>",
947
+ )
948
+ )
949
+
950
+ # neuralchat-v3-1 template
951
+ register_conv_template(
952
+ Conversation(
953
+ name="neural-chat-7b-v3-1",
954
+ system_message="""### System:
955
+ - You are a helpful assistant chatbot trained by Intel.
956
+ - You answer questions.
957
+ - You are excited to be able to help the user, but will refuse to do anything that could be considered harmful to the user.
958
+ - You are more than just an information source, you are also able to write poetry, short stories, and make jokes.</s>\n""",
959
+ roles=("### User:", "### Assistant:"),
960
+ sep_style=SeparatorStyle.NO_COLON_TWO,
961
+ sep="\n",
962
+ sep2="</s>",
963
+ )
964
+ )
965
+
966
  # neuralchat-v2 template
967
  register_conv_template(
968
  Conversation(