From 2803587c7772b30f145f6f31d49a4cfd16377e99 Mon Sep 17 00:00:00 2001 From: Atridad Lahiji Date: Tue, 5 Mar 2024 01:39:39 -0600 Subject: [PATCH] Update image generation parameters --- lib/replicate.go | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/lib/replicate.go b/lib/replicate.go index ebe8dd6..bf511fd 100644 --- a/lib/replicate.go +++ b/lib/replicate.go @@ -72,11 +72,10 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er "width": 1024, "height": 1024, "prompt": prompt, - "scheduler": "K_EULER", - "refine": "expert_ensemble_refiner", "num_outputs": 1, - "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, text, BadDream, lowres, low resolution, mutated body parts, extra limbs, mutated body parts, inaccurate hands, too many hands, deformed fingers, too many fingers, deformed eyes, deformed faces, unrealistic faces", - "num_inference_steps": 35, + "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, BadDream, lowres, low resolution, mutated body parts, extra limbs, mutated body parts, inaccurate hands, too many hands, deformed fingers, too many fingers, deformed eyes, deformed faces, unrealistic faces", + "num_inference_steps": 50, + "apply_watermark": false, "disable_safety_checker": true, } webhook := replicate.Webhook{ @@ -84,7 +83,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/playground-v2.5-1024px-aesthetic:419269784d9e00c56e5b09747cfc059a421e0c044d5472109e129b746508c365", input, &webhook) if predictionError != nil { return nil, predictionError