create batch size latents

2023-07-04 17:27:10 -05:00 · 2023-07-04 17:27:10 -05:00 · 10050263bf
parent 89dbf81335
commit 10050263bf
1 changed files with 1 additions and 1 deletions
--- a/api/onnx_web/chain/source_txt2img.py
+++ b/api/onnx_web/chain/source_txt2img.py
@ -43,7 +43,7 @@ class SourceTxt2ImgStage(BaseStage):

        prompt_pairs, loras, inversions = parse_prompt(params)

-        latents = get_latents_from_seed(params.seed, size)
+        latents = get_latents_from_seed(params.seed, size, params.batch)
        pipe_type = params.get_valid_pipeline("txt2img")
        pipe = load_pipeline(
            server,