move prompts to json file

This commit is contained in:
Bruce MacDonald
2023-06-26 14:03:49 -04:00
parent 0fc80e0f21
commit 93e0ffc9e0
2 changed files with 15 additions and 40 deletions

View File

@@ -1,45 +1,8 @@
from difflib import SequenceMatcher
import json
model_prompts = {
"alpaca": """Below is an instruction that describes a task. Write a response that appropriately completes the request.
### Instruction:
{prompt}
### Response:
""",
"oasst": "<|prompter|>{prompt}<|endoftext|><|assistant|>",
"vicuna": """A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.
USER: {prompt}
ASSISTANT:""",
"hermes": """### Instruction:
{prompt}
### Response:
""",
"gpt4": """### Instruction:
{prompt}
### Response:
""",
"qlora": """### Human: {prompt}
### Assistant:""",
"tulu": """<|user|>
{prompt}
<|assistant|>
(include newline)""",
"wizardlm-7b": """{prompt}
### Response:""",
"wizardlm-13b": """{prompt}
### Response:""",
"wizardlm-30b": """{prompt}
### Response:""",
}
with open("model_prompts.json", "r") as f:
model_prompts = json.load(f)
def template(model, prompt):