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

Update image generation model in replicate.go

Browse source
This commit is contained in:
Atridad Lahiji 2024-03-09 23:13:38 -06:00
parent 3c9a431ab7
commit adac24372e
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/replicate.go
View file

@ -73,8 +73,6 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
"height": 1024,
"prompt": prompt,
"num_outputs": 1,
"num_inference_steps": 60,
"apply_watermark": false,
"disable_safety_checker": true,
}
webhook := replicate.Webhook{
@ -82,7 +80,7 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
Events: []replicate.WebhookEventType{"start", "completed"},
}
prediction, predictionError := client.Run(context.Background(), "lucataco/playground-v2.5-1024px-aesthetic:419269784d9e00c56e5b09747cfc059a421e0c044d5472109e129b746508c365", input, &webhook)
prediction, predictionError := client.Run(context.Background(), "lucataco/sdxl-lightning-4stepc:727e49a643e999d602a896c774a0658ffefea21465756a6ce24b7ea4165eba6a", input, &webhook)
if predictionError != nil {
return nil, predictionError