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": "[INST] {prompt} [/INST]", + "prompt": ReplicatePromptPrefix + prompt, + "max_new_tokens": 4096, } webhook := replicate.Webhook{