1
0
Fork 0

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

View File

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