apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

revert back to NL means for panorama only

This commit is contained in:
Sean Sube 2023-12-29 22:53:04 -06:00
parent 750a49a907
commit 05ab396b2a
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/diffusers/run.py
View File

@ -5,7 +5,7 @@ from typing import Any, List, Optional
from PIL import Image, ImageOps from PIL import Image, ImageOps
from ..chain import ( from ..chain import (
BlendDenoiseLocalStdStage, BlendDenoiseFastNLMeansStage,
BlendImg2ImgStage, BlendImg2ImgStage,
BlendMaskStage, BlendMaskStage,
ChainPipeline, ChainPipeline,
@ -78,9 +78,9 @@ def run_txt2img_pipeline(
# apply upscaling and correction, before highres # apply upscaling and correction, before highres
highres_size = get_highres_tile(server, params, highres, tile_size) highres_size = get_highres_tile(server, params, highres, tile_size)
if params.is_xl(): if params.is_panorama():
chain.stage( chain.stage(
BlendDenoiseLocalStdStage(), BlendDenoiseFastNLMeansStage(),
StageParams(tile_size=highres_size), StageParams(tile_size=highres_size),
) )