From 05ab396b2ad668c3c5c678e40d8b2261b0fdbb8a Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Fri, 29 Dec 2023 22:53:04 -0600
Subject: [PATCH] revert back to NL means for panorama only

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

diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
index 48269fd4..a3737d06 100644
--- a/api/onnx_web/diffusers/run.py
+++ b/api/onnx_web/diffusers/run.py
@@ -5,7 +5,7 @@ from typing import Any, List, Optional
 from PIL import Image, ImageOps
 
 from ..chain import (
-    BlendDenoiseLocalStdStage,
+    BlendDenoiseFastNLMeansStage,
     BlendImg2ImgStage,
     BlendMaskStage,
     ChainPipeline,
@@ -78,9 +78,9 @@ def run_txt2img_pipeline(
 
     # apply upscaling and correction, before highres
     highres_size = get_highres_tile(server, params, highres, tile_size)
-    if params.is_xl():
+    if params.is_panorama():
         chain.stage(
-            BlendDenoiseLocalStdStage(),
+            BlendDenoiseFastNLMeansStage(),
             StageParams(tile_size=highres_size),
         )