[diffusion] app: add ComfyUI plugin support for SGLang-Diffusion (#15271)

Co-authored-by: niehen6174 <niehen.6174@gmail.com>
Co-authored-by: Mick <mickjagger19@icloud.com>
Co-authored-by: niehen6174 <nihen6174@gmail.com>
This commit is contained in:
WenhaoZhang
2026-01-12 21:58:16 +08:00
committed by GitHub
co-authored by niehen6174 Mick niehen6174
parent 9a628744fc
commit 76f69b7753
30 changed files with 3688 additions and 12 deletions
@@ -265,6 +265,11 @@ class ServerArgs:
pipeline_config: PipelineConfig = field(default_factory=PipelineConfig, repr=False)
# Pipeline override
pipeline_class_name: str | None = (
None # Override pipeline class from model_index.json
)
# LoRA parameters
# (Wenxuan) prefer to keep it here instead of in pipeline config to not make it complicated.
lora_path: str | None = None
@@ -285,6 +290,9 @@ class ServerArgs:
use_fsdp_inference: bool = False
pin_cpu_memory: bool = True
# ComfyUI integration
comfyui_mode: bool = False
# STA (Sliding Tile Attention) parameters
mask_strategy_file_path: str | None = None
STA_mode: STA_Mode = STA_Mode.STA_INFERENCE