additional props
This commit is contained in:
parent
69da29eecd
commit
bba02f0303
1 changed files with 4 additions and 1 deletions
|
@ -59,6 +59,9 @@ func ReplicateImageGeneration(prompt string) (*bytes.Buffer, error) {
|
||||||
|
|
||||||
input := replicate.PredictionInput{
|
input := replicate.PredictionInput{
|
||||||
"prompt": prompt,
|
"prompt": prompt,
|
||||||
|
"refiner": "expert_ensemble_refiner",
|
||||||
|
"num_inference_steps": 69,
|
||||||
|
"disable_safety_checker": true,
|
||||||
}
|
}
|
||||||
webhook := replicate.Webhook{
|
webhook := replicate.Webhook{
|
||||||
URL: "https://example.com/webhook",
|
URL: "https://example.com/webhook",
|
||||||
|
|
Loading…
Add table
Reference in a new issue