From ddecab5cecf5e27c96a82dba6df5b333793c6326 Mon Sep 17 00:00:00 2001
From: Atridad Lahiji <me@atri.dad>
Date: Mon, 26 Feb 2024 00:42:07 -0700
Subject: [PATCH] Fixed chat

---
 lib/replicate.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/lib/replicate.go b/lib/replicate.go
index 08e83ec..cd86dbe 100644
--- a/lib/replicate.go
+++ b/lib/replicate.go
@@ -22,9 +22,8 @@ func ReplicateTextGeneration(prompt string) (string, error) {
 	}
 
 	input := replicate.PredictionInput{
-		"prompt":          ReplicatePromptPrefix + prompt,
-		"max_new_tokens":  4096,
-		"prompt_template": "<s>[INST] {prompt} [/INST]",
+		"prompt":         ReplicatePromptPrefix + prompt,
+		"max_new_tokens": 4096,
 	}
 
 	webhook := replicate.Webhook{