@@ -57,7 +57,8 @@ MODELS=`[
57
57
"repetition_penalty": 1.2,
58
58
"top_k": 50,
59
59
"truncate": 3072,
60
- "max_new_tokens": 1024
60
+ "max_new_tokens": 1024,
61
+ "stop" : ["</s>", " </s><s>[INST] "]
61
62
}
62
63
},
63
64
{
@@ -116,7 +117,8 @@ MODELS=`[
116
117
"repetition_penalty": 1.2,
117
118
"top_k": 50,
118
119
"truncate": 4096,
119
- "max_new_tokens": 4096
120
+ "max_new_tokens": 4096,
121
+ "stop": [" </s><s>[INST] "]
120
122
}
121
123
},
122
124
{
@@ -218,19 +220,18 @@ OLD_MODELS=`[
218
220
{"name": "tiiuae/falcon-180B-chat"}
219
221
]`
220
222
221
- TASK_MODEL='mistralai/Mistral-7B-Instruct-v0.2'
222
- # TASK_MODEL=`{
223
- # "name": "mistralai/Mistral-7B-Instruct-v0.2",
224
- # "chatPromptTemplate" : "<s>{{#each messages}}{{#ifUser}}[INST] {{#if @first}}{{#if @root.preprompt}}{{@root.preprompt}}\n{{/if}}{{/if}}{{content}} [/INST]{{/ifUser}}{{#ifAssistant}}{{content}}</s>{{/ifAssistant}}{{/each}}",
225
- # "parameters": {
226
- # "temperature": 0.1,
227
- # "top_p": 0.95,
228
- # "repetition_penalty": 1.2,
229
- # "top_k": 50,
230
- # "truncate": 3072,
231
- # "max_new_tokens": 1024,
232
- # "stop": ["</s>"]
233
- # }}`
223
+ TASK_MODEL=`{
224
+ "name": "mistralai/Mistral-7B-Instruct-v0.2",
225
+ "chatPromptTemplate" : "<s>{{#each messages}}{{#ifUser}}[INST] {{#if @first}}{{#if @root.preprompt}}{{@root.preprompt}}\n{{/if}}{{/if}}{{content}} [/INST]{{/ifUser}}{{#ifAssistant}}{{content}}</s>{{/ifAssistant}}{{/each}}",
226
+ "parameters": {
227
+ "temperature": 0.1,
228
+ "top_p": 0.95,
229
+ "repetition_penalty": 1.2,
230
+ "top_k": 50,
231
+ "truncate": 3072,
232
+ "max_new_tokens": 1024,
233
+ "stop": ["</s>"]
234
+ }}`
234
235
235
236
APP_BASE="/chat"
236
237
PUBLIC_ORIGIN=https://huggingface.co
0 commit comments