fix(api): make sure stage params are declared for all txt2img pipelines

ssube · Nov 16, 2023 · b31227e · b31227e
1 parent 5968577
commit b31227e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,