Spaces:
Runtime error
Runtime error
Update conversation.py
Browse files- 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(
|