atridad/himbot
1
0
Fork 0
Code Projects Releases Packages 1 Activity Actions

larger allowed output

Browse source
This commit is contained in:
Atridad Lahiji 2024-01-10 01:03:25 -07:00
parent b3e6291ad6
commit 7a9e1da0e2
No known key found for this signature in database
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
lib/replicate.go
View file

@ -18,7 +18,8 @@ func ReplicateTextGeneration(prompt string) (string, error) {
} }
input := replicate.PredictionInput{ input := replicate.PredictionInput{
"prompt": prompt, "prompt": prompt,
"max_new_tokens": 4096,
} }
webhook := replicate.Webhook{ webhook := replicate.Webhook{
URL: "https://example.com/webhook", URL: "https://example.com/webhook",