diff --git a/orchestrators/imajin-pipeline/pyproject.toml b/orchestrators/imajin-pipeline/pyproject.toml index d800fb6b..6d1856d4 100644 --- a/orchestrators/imajin-pipeline/pyproject.toml +++ b/orchestrators/imajin-pipeline/pyproject.toml @@ -71,7 +71,10 @@ pulid = [ "onnxruntime-gpu>=1.15.0", # InsightFace GPU backend "model-boss-loaders[pulid]>=1.0.0", # PuLID loader ] -all = ["imajin-pipeline[quality,moderation,anatomy_fix,watermark_removal,controlnet,flux,pulid]"] +background_removal = [ + "rembg>=2.0.50", # U2Net-based background removal +] +all = ["imajin-pipeline[quality,moderation,anatomy_fix,watermark_removal,controlnet,flux,pulid,background_removal]"] dev = ["pytest>=7.0", "pytest-asyncio>=0.21", "mypy>=1.0"] [tool.hatch.build.targets.wheel]