From 33542f2f1b466b2a0441d756691c7679277ec6da Mon Sep 17 00:00:00 2001 From: atridadl Date: Mon, 29 Jan 2024 09:43:11 -0700 Subject: [PATCH] Update prompt in ReplicateTextGeneration function --- lib/replicate.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/replicate.go b/lib/replicate.go index bf81ef9..8087be5 100644 --- a/lib/replicate.go +++ b/lib/replicate.go @@ -20,7 +20,7 @@ func ReplicateTextGeneration(prompt string) (string, error) { } input := replicate.PredictionInput{ - "prompt": prompt, + "prompt": "Respond to the following prompt as the helpful but sarcastic and witty discord assistant called Himbot: " + prompt, "max_new_tokens": 4096, "prompt_template": "[INST] {prompt} [/INST]", }