From 0f17c439e7b1eb836e468fc6d9dab1cc290063c2 Mon Sep 17 00:00:00 2001
From: Atridad Lahiji <me@atri.dad>
Date: Sat, 30 Mar 2024 02:17:49 -0600
Subject: [PATCH] Model change

---
 lib/replicate.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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