fix(api): pass batch size and eta params to txt2img stage

ssube · Jul 4, 2023 · 89dbf81 · 89dbf81
1 parent 17b90ef
commit 89dbf81
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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,
             )