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]", }