put prompt filters behind a feature flag for now
This commit is contained in:
parent
bc2eeb8503
commit
3fea317a43
|
@ -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(
|
||||||
|
|
Loading…
Reference in New Issue