additional props

This commit is contained in:
Atridad Lahiji 2024-01-10 00:46:10 -07:00
parent 69da29eecd
commit bba02f0303
No known key found for this signature in database

View file

@ -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",