Mishig commited on
Commit
22e7bfa
1 Parent(s): 77399ca

Update readme to use `meta-llama/Llama-2-70b-chat-hf` (#723)

Browse files

* Update readme to use `meta-llama/Llama-2-70b-chat-hf`

* add `mistralai/Mistral-7B-Instruct-v0.2` instead

Files changed (1) hide show
  1. README.md +17 -20
README.md CHANGED
@@ -174,36 +174,33 @@ You can customize the parameters passed to the model or even use a new model by
174
  ```env
175
  MODELS=`[
176
  {
177
- "name": "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
178
- "datasetName": "OpenAssistant/oasst1",
179
- "description": "A good alternative to ChatGPT",
180
- "websiteUrl": "https://open-assistant.io",
181
- "userMessageToken": "<|prompter|>", # This does not need to be a token, can be any string
182
- "assistantMessageToken": "<|assistant|>", # This does not need to be a token, can be any string
183
- "userMessageEndToken": "<|endoftext|>", # Applies only to user messages. Can be any string.
184
- "assistantMessageEndToken": "<|endoftext|>", # Applies only to assistant messages. Can be any string.
185
- "preprompt": "Below are a series of dialogues between various people and an AI assistant. The AI tries to be helpful, polite, honest, sophisticated, emotionally aware, and humble but knowledgeable. The assistant is happy to help with almost anything and will do its best to understand exactly what is needed. It also tries to avoid giving false or misleading information, and it caveats when it isn't entirely sure about the right answer. That said, the assistant is practical and really does its best, and doesn't let caution get too much in the way of being useful.\n-----\n",
 
 
 
 
 
 
186
  "promptExamples": [
187
  {
188
  "title": "Write an email from bullet list",
189
  "prompt": "As a restaurant owner, write a professional email to the supplier to get these products every week: \n\n- Wine (x10)\n- Eggs (x24)\n- Bread (x12)"
190
  }, {
191
  "title": "Code a snake game",
192
- "prompt": "Code a basic snake game in python and give explanations for each step."
193
  }, {
194
  "title": "Assist in a task",
195
  "prompt": "How do I make a delicious lemon cheesecake?"
196
  }
197
- ],
198
- "parameters": {
199
- "temperature": 0.9,
200
- "top_p": 0.95,
201
- "repetition_penalty": 1.2,
202
- "top_k": 50,
203
- "truncate": 1000,
204
- "max_new_tokens": 1024,
205
- "stop": ["<|endoftext|>"] # This does not need to be tokens, can be any list of strings
206
- }
207
  }
208
  ]`
209
 
 
174
  ```env
175
  MODELS=`[
176
  {
177
+ "name": "mistralai/Mistral-7B-Instruct-v0.2",
178
+ "displayName": "mistralai/Mistral-7B-Instruct-v0.2",
179
+ "description": "Mistral 7B is a new Apache 2.0 model, released by Mistral AI that outperforms Llama2 13B in benchmarks.",
180
+ "websiteUrl": "https://mistral.ai/news/announcing-mistral-7b/",
181
+ "preprompt": "",
182
+ "chatPromptTemplate" : "<s>{{#each messages}}{{#ifUser}}[INST] {{#if @first}}{{#if @root.preprompt}}{{@root.preprompt}}\n{{/if}}{{/if}}{{content}} [/INST]{{/ifUser}}{{#ifAssistant}}{{content}}</s>{{/ifAssistant}}{{/each}}",
183
+ "parameters": {
184
+ "temperature": 0.3,
185
+ "top_p": 0.95,
186
+ "repetition_penalty": 1.2,
187
+ "top_k": 50,
188
+ "truncate": 3072,
189
+ "max_new_tokens": 1024,
190
+ "stop": ["</s>"]
191
+ },
192
  "promptExamples": [
193
  {
194
  "title": "Write an email from bullet list",
195
  "prompt": "As a restaurant owner, write a professional email to the supplier to get these products every week: \n\n- Wine (x10)\n- Eggs (x24)\n- Bread (x12)"
196
  }, {
197
  "title": "Code a snake game",
198
+ "prompt": "Code a basic snake game in python, give explanations for each step."
199
  }, {
200
  "title": "Assist in a task",
201
  "prompt": "How do I make a delicious lemon cheesecake?"
202
  }
203
+ ]
 
 
 
 
 
 
 
 
 
204
  }
205
  ]`
206