From 5be11658581deedc489ed2308f11f6ca423cca9b Mon Sep 17 00:00:00 2001 From: Atridad Lahiji Date: Thu, 22 Feb 2024 15:33:33 -0700 Subject: [PATCH] Made image gen WAY faster --- lib/replicate.go | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/lib/replicate.go b/lib/replicate.go index 9c67f0b..37e2af8 100644 --- a/lib/replicate.go +++ b/lib/replicate.go @@ -113,9 +113,14 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er } input := replicate.PredictionInput{ + "width": 1024, + "height": 1024, "prompt": prompt, - "refiner": "expert_ensemble_refiner", - "num_inference_steps": 30, + "scheduler": "K_EULER", + "num_outputs": 1, + "guidance_scale": 0, + "negative_prompt": "worst quality, low quality", + "num_inference_steps": 4, "disable_safety_checker": true, } webhook := replicate.Webhook{ @@ -123,7 +128,7 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er Events: []replicate.WebhookEventType{"start", "completed"}, } - prediction, predictionError := client.Run(context.Background(), "stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b", input, &webhook) + prediction, predictionError := client.Run(context.Background(), "lucataco/sdxl-lightning-4step:727e49a643e999d602a896c774a0658ffefea21465756a6ce24b7ea4165eba6a", input, &webhook) if predictionError != nil { return nil, predictionError