From b65e4e7cbeb213fab5e152f90d0575381c59c253 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Mon, 20 Feb 2023 21:57:46 -0600
Subject: [PATCH] use batch size when generating latents

---
 api/onnx_web/diffusion/run.py | 2 +-
 gui/src/client.ts             | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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<string>) {
  */
 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));