From 10050263bf474ca93a62b055325b71a55809e1d0 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 4 Jul 2023 17:27:10 -0500 Subject: [PATCH] create batch size latents --- api/onnx_web/chain/source_txt2img.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index 728d2633..ebc8250f 100644 --- 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,