lianglv commited on
Commit
45c5a5e
1 Parent(s): 8073951

Update conversation.py

Browse files
Files changed (1) hide show
  1. conversation.py +12 -0
conversation.py CHANGED
@@ -930,6 +930,18 @@ register_conv_template(
930
  )
931
  )
932
 
 
 
 
 
 
 
 
 
 
 
 
 
933
 
934
  # neuralchat-v3 template
935
  register_conv_template(
 
930
  )
931
  )
932
 
933
+ # Mistral template
934
+ # source: https://docs.mistral.ai/llm/mistral-instruct-v0.1#chat-template
935
+ register_conv_template(
936
+ Conversation(
937
+ name="mistral",
938
+ system_template="[INST]{system_message}\n",
939
+ roles=("[INST]", "[/INST]"),
940
+ sep_style=SeparatorStyle.LLAMA2,
941
+ sep=" ",
942
+ sep2="</s>",
943
+ )
944
+ )
945
 
946
  # neuralchat-v3 template
947
  register_conv_template(