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

Model change

Browse source
This commit is contained in:
Atridad Lahiji 2024-03-30 02:17:49 -06:00
parent 55c7eb03be
commit 0f17c439e7
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

@ -72,6 +72,7 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
"width": 1024,
"height": 1024,
"prompt": prompt,
"refine": "expert_ensemble_refiner",
"num_outputs": 1,
"disable_safety_checker": true,
}
@ -80,7 +81,7 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
Events: []replicate.WebhookEventType{"start", "completed"},
}
prediction, predictionError := client.Run(context.Background(), "lucataco/sdxl-lightning-4stepc:727e49a643e999d602a896c774a0658ffefea21465756a6ce24b7ea4165eba6a", input, &webhook)
prediction, predictionError := client.Run(context.Background(), "stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b", input, &webhook)
if predictionError != nil {
return nil, predictionError