Update replicate.go

This commit is contained in:
Atridad Lahiji 2024-02-19 12:33:23 -07:00 committed by GitHub
parent 041552cf77
commit dcc57ccb74
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -115,7 +115,7 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
input := replicate.PredictionInput{
"prompt": prompt,
"refiner": "expert_ensemble_refiner",
"num_inference_steps": 69,
"num_inference_steps": 30,
"disable_safety_checker": true,
}
webhook := replicate.Webhook{