Browse Source

remove ggml template

Bruce MacDonald 1 year ago
parent
commit
bf730ac6c5
2 changed files with 1 additions and 1 deletions
  1. 1 0
      ollama.py
  2. 0 1
      template.py

+ 1 - 0
ollama.py

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

+ 0 - 1
template.py

@@ -2,7 +2,6 @@ from difflib import SequenceMatcher
 
 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",
-    "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",
     "hermes": "### Instruction:\n{prompt}\n\n### Response:\n",
     "oasst": "{prompt}",