baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

remove ggml template

Browse source
This commit is contained in:
Bruce MacDonald 2023-06-27 15:29:47 -04:00
parent ecfb4abafb
commit bf730ac6c5
2 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
ollama.py
View file

@ -132,6 +132,7 @@ def generate_route_handler():
data = request.get_json() data = request.get_json()
model = data.get("model") model = data.get("model")
prompt = data.get("prompt") prompt = data.get("prompt")
prompt = template(model, prompt)
if not model: if not model:
return Response("Model is required", status=400) return Response("Model is required", status=400)
if not prompt: if not prompt:

1
template.py
View file

@ -2,7 +2,6 @@ from difflib import SequenceMatcher
model_prompts = { model_prompts = {
"alpaca": "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n{prompt}\n\n### Response:\n\n", "alpaca": "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n{prompt}\n\n### Response:\n\n",
"ggml": "A chat between a curious human and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the human's questions.\n\n### Human: Hello, Assistant.\n### Assistant: Hello. How may I help you today?\n### Human: ${prompt}",
"gpt4": "### Instruction:\n{prompt}\n\n### Response:\n", "gpt4": "### Instruction:\n{prompt}\n\n### Response:\n",
"hermes": "### Instruction:\n{prompt}\n\n### Response:\n", "hermes": "### Instruction:\n{prompt}\n\n### Response:\n",
"oasst": "{prompt}", "oasst": "{prompt}",