diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py index a5417e84..c96af7e1 100644 --- a/api/onnx_web/diffusion/run.py +++ b/api/onnx_web/diffusion/run.py @@ -28,7 +28,7 @@ def run_txt2img_pipeline( outputs: List[str], upscale: UpscaleParams, ) -> None: - latents = get_latents_from_seed(params.seed, size) + latents = get_latents_from_seed(params.seed, size, batch=params.batch) pipe = load_pipeline( server, OnnxStableDiffusionPipeline, diff --git a/gui/src/client.ts b/gui/src/client.ts index 6d094552..aee39029 100644 --- a/gui/src/client.ts +++ b/gui/src/client.ts @@ -280,6 +280,7 @@ export function makeApiUrl(root: string, ...path: Array) { */ export function makeImageURL(root: string, type: string, params: BaseImgParams): URL { const url = makeApiUrl(root, type); + url.searchParams.append('batch', params.batch.toFixed(FIXED_INTEGER)); url.searchParams.append('cfg', params.cfg.toFixed(FIXED_FLOAT)); url.searchParams.append('eta', params.eta.toFixed(FIXED_FLOAT)); url.searchParams.append('steps', params.steps.toFixed(FIXED_INTEGER));