Skip to content

Commit da4edc5

Browse files
authored
Use a low temperature mistral for tasks (#652)
1 parent bdb1788 commit da4edc5

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

.env.template

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,18 @@ MODELS=`[
213213

214214
OLD_MODELS=`[{"name":"bigcode/starcoder"}, {"name":"OpenAssistant/oasst-sft-6-llama-30b-xor"}, {"name":"HuggingFaceH4/zephyr-7b-alpha"}]`
215215

216-
TASK_MODEL='mistralai/Mistral-7B-Instruct-v0.2'
216+
TASK_MODEL=`{
217+
"name": "mistralai/Mistral-7B-Instruct-v0.2",
218+
"chatPromptTemplate" : "<s>{{#each messages}}{{#ifUser}}[INST] {{#if @first}}{{#if @root.preprompt}}{{@root.preprompt}}\n{{/if}}{{/if}}{{content}} [/INST]{{/ifUser}}{{#ifAssistant}}{{content}}</s>{{/ifAssistant}}{{/each}}",
219+
"parameters": {
220+
"temperature": 0.1,
221+
"top_p": 0.95,
222+
"repetition_penalty": 1.2,
223+
"top_k": 50,
224+
"truncate": 3072,
225+
"max_new_tokens": 1024,
226+
"stop": ["</s>"]
227+
}}`
217228

218229

219230
APP_BASE="/chat"

0 commit comments

Comments
 (0)