apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-07-04 17:24:37 -05:00
parent 17b90ef2d7
commit 89dbf81335
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/chain/source_txt2img.py
View File

@ -65,7 +65,9 @@ class SourceTxt2ImgStage(BaseStage):
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 @@ class SourceTxt2ImgStage(BaseStage):
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,
)