From b31227ecb3d4156088c5ec1c134a837cb9c071bd Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Wed, 15 Nov 2023 23:14:48 -0600
Subject: [PATCH] fix(api): make sure stage params are declared for all txt2img
 pipelines

---
 api/onnx_web/diffusers/run.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
index 11594468..45ef3881 100644
--- a/api/onnx_web/diffusers/run.py
+++ b/api/onnx_web/diffusers/run.py
@@ -63,8 +63,9 @@ def run_txt2img_pipeline(
 
     # apply upscaling and correction, before highres
     highres_size = params.unet_tile
+    stage = StageParams(tile_size=highres_size)
+
     if params.is_panorama():
-        stage = StageParams(tile_size=highres_size)
         chain.stage(
             BlendDenoiseStage(),
             stage,