atridad/himbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update replicate.go

Browse Source
This commit is contained in:
Atridad Lahiji
2024-01-25 10:23:18 -07:00
committed by GitHub
parent 6783d82bf2
commit 939710133d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/replicate.go
View File

@ -27,7 +27,7 @@ func ReplicateTextGeneration(prompt string) (string, error) {
Events: []replicate.WebhookEventType{"start", "completed"}, Events: []replicate.WebhookEventType{"start", "completed"},
} }
prediction, predictionError := client.Run(context.Background(), "meta/llama-2-70b-chat:2d19859030ff705a87c746f7e96eea03aefb71f166725aee39692f1476566d48", input, &webhook) prediction, predictionError := client.Run(context.Background(), "mistralai/mistral-7b-instruct-v0.2:79052a3adbba8116ebc6697dcba67ad0d58feff23e7aeb2f103fc9aa545f9269", input, &webhook)
if predictionError != nil { if predictionError != nil {
return "", predictionError return "", predictionError