From bba02f0303ddf5a29e7236a0676abe389c86885f Mon Sep 17 00:00:00 2001 From: atridadl Date: Wed, 10 Jan 2024 00:46:10 -0700 Subject: [PATCH] additional props --- lib/replicate.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/lib/replicate.go b/lib/replicate.go index 77deb5f..35c77cc 100644 --- a/lib/replicate.go +++ b/lib/replicate.go @@ -58,7 +58,10 @@ func ReplicateImageGeneration(prompt string) (*bytes.Buffer, error) { } input := replicate.PredictionInput{ - "prompt": prompt, + "prompt": prompt, + "refiner": "expert_ensemble_refiner", + "num_inference_steps": 69, + "disable_safety_checker": true, } webhook := replicate.Webhook{ URL: "https://example.com/webhook",