added new models + prompt formats

predictionguard · Mar 19, 2024 · 2f1a7d3 · 2f1a7d3
1 parent 55b4b1f
commit 2f1a7d3
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/fern/docs/pages/models/details.mdx b/fern/docs/pages/models/details.mdx
@@ -20,11 +20,12 @@ Open access models are amazing these days! Each of these models was trained by a
 | Model Name                   | Type            | Use Case                                                | Prompt Format                      | Context Length | More Info                                                               |
 | ---------------------------- | --------------- | ------------------------------------------------------- | ---------------------------------- | -------------- | ----------------------------------------------------------------------- |
 | Nous-Hermes-Llama2-13B       | Text Generation | Generating output in response to arbitrary instructions | [Alpaca](prompts#alpaca)           | 4096           | [link](https://huggingface.co/NousResearch/Nous-Hermes-Llama2-13b)      |
-| Nous-Hermes-2-SOLAR-10.7B    | Chat            | Instruction following or chat-like applications         | [ChatML](prompts#chatml)           | 4096           | [link](https://huggingface.co/NousResearch/Nous-Hermes-2-SOLAR-10.7B)   |
+| Hermes-2-Pro-Mistral-7B      | Chat            | Instruction following or chat-like applications         | [ChatML](prompts#chatml)           | 4096           | [link](https://huggingface.co/NousResearch/Hermes-2-Pro-Mistral-7B)     |
 | Neural-Chat-7B               | Chat            | Instruction following or chat-like applications         | [Neural Chat](prompts#neural-chat) | 4096           | [link](https://huggingface.co/Intel/neural-chat-7b-v3-1)                |
 | Yi-34B-Chat                  | Chat            | Instruction following in English or Chinese             | [ChatML](prompts#chatml)           | 2048           | [link](https://huggingface.co/01-ai/Yi-34B-Chat)                        |
-| sqlcoder-34b-alpha           | Code Generation | Generating SQL queries from natural language prompts    | [SQLCoder](prompts#sqlcoder)       | 4096           | [link](https://huggingface.co/defog/sqlcoder-34b-alpha)                 |
+| sqlcoder-34b-alpha           | Code Generation | Generating SQL queries from natural language prompts    | [SQLCoder](prompts#sqlcoder-34b)   | 4096           | [link](https://huggingface.co/defog/sqlcoder-34b-alpha)                 |
 | deepseek-coder-6.7b-instruct | Code Generation | Generating computer code or answering tech questions    | [Deepseek](prompts#deepseek)       | 4096           | [link](https://huggingface.co/deepseek-ai/deepseek-coder-6.7b-instruct) |
+| sqlcoder-7b-2                | Code Generation | Generating SQL queries from natural language prompts    | [SQLCoder](prompts#sqlcoder-7b)    | 4096           | [link](https://huggingface.co/defog/sqlcoder-7b-2)                      |
 
 ### Other models available
 

diff --git a/fern/docs/pages/models/prompts.mdx b/fern/docs/pages/models/prompts.mdx
@@ -71,7 +71,7 @@ For prompts where context is injected:
 <|im_start|>assistant<|im_end|>
 ```
 
-## SQLCoder
+## SQLCoder-34b
 
 (Replace the portions of the prompt below in curly braces `{...}` with the appropriate information, and do not keep the curly braces)
 
@@ -91,6 +91,23 @@ This query will run on a database whose schema is represented in this string:
 Given the database schema, here is the SQL query that answers `{question}`:
 ```
 
+## SQLCoder-7b
+
+(Replace the portions of the prompt below in curly braces `{...}` with the appropriate information, and do not keep the curly braces)
+
+```
+### Task
+Generate a SQL query to answer [QUESTION]{user_question}[/QUESTION]
+
+### Database Schema
+The query will run on a database with the following schema:
+{table_metadata_string_DDL_statements}
+
+### Answer
+Given the database schema, here is the SQL query that [QUESTION]{user_question}[/QUESTION]
+[SQL]
+```
+
 ## Deepseek
 
 (Replace the portions of the prompt below in curly braces `{...}` with the appropriate information, and do not keep the curly braces)