From 7a9e1da0e29b72fd98266599058dd621e680e8ae Mon Sep 17 00:00:00 2001 From: atridadl Date: Wed, 10 Jan 2024 01:03:25 -0700 Subject: [PATCH] larger allowed output --- lib/replicate.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/replicate.go b/lib/replicate.go index 35c77cc..e251791 100644 --- a/lib/replicate.go +++ b/lib/replicate.go @@ -18,7 +18,8 @@ func ReplicateTextGeneration(prompt string) (string, error) { } input := replicate.PredictionInput{ - "prompt": prompt, + "prompt": prompt, + "max_new_tokens": 4096, } webhook := replicate.Webhook{ URL: "https://example.com/webhook",