diff --git a/lib/replicate.go b/lib/replicate.go index 8a6cf27..3add4fa 100644 --- a/lib/replicate.go +++ b/lib/replicate.go @@ -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