apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

put prompt filters behind a feature flag for now

This commit is contained in:
Sean Sube 2024-02-11 16:25:34 -06:00
parent bc2eeb8503
commit 3fea317a43
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -9,6 +9,7 @@ from ..chain import (
BlendMaskStage, BlendMaskStage,
ChainPipeline, ChainPipeline,
SourceTxt2ImgStage, SourceTxt2ImgStage,
TextPromptStage,
UpscaleOutpaintStage, UpscaleOutpaintStage,
) )
from ..chain.highres import stage_highres from ..chain.highres import stage_highres
@ -76,6 +77,13 @@ def run_txt2img_pipeline(
# prepare the chain pipeline and first stage # prepare the chain pipeline and first stage
chain = ChainPipeline() chain = ChainPipeline()
if server.has_feature("prompt-filter"):
chain.stage(
TextPromptStage(),
StageParams(),
)
chain.stage( chain.stage(
SourceTxt2ImgStage(), SourceTxt2ImgStage(),
StageParams( StageParams(