diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index b5d16147a..728d26332 100644 --- a/api/onnx_web/chain/source_txt2img.py +++ b/api/onnx_web/chain/source_txt2img.py @@ -65,7 +65,9 @@ def run( guidance_scale=params.cfg, latents=latents, negative_prompt=params.negative_prompt, + num_images_per_prompt=params.batch, num_inference_steps=params.steps, + eta=params.eta, callback=callback, ) else: @@ -84,7 +86,9 @@ def run( guidance_scale=params.cfg, latents=latents, negative_prompt=params.negative_prompt, + num_images_per_prompt=params.batch, num_inference_steps=params.steps, + eta=params.eta, callback=callback, )