minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] fix: fix bugs to let LTX-2 pipeline support latest Sglang Args pipelines (#19295)

Browse Source
This commit is contained in:
GMI Xiao Jin
2026-02-25 17:30:36 +08:00
committed by GitHub
parent 671b595570
commit c4ef33862b
3 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/configs/pipeline_configs/base.py
+1
View File
@@ -153,6 +153,7 @@ class PipelineConfig:
"""The base configuration class for a generation pipeline.""" """The base configuration class for a generation pipeline."""
task_type: ModelTaskType = ModelTaskType.I2I task_type: ModelTaskType = ModelTaskType.I2I
skip_input_image_preprocess: bool = False
model_path: str = "" model_path: str = ""
pipeline_config_path: str | None = None pipeline_config_path: str | None = None
python/sglang/multimodal_gen/configs/pipeline_configs/ltx_2.py
+2 -1
View File
@@ -115,7 +115,8 @@ def _gemma_postprocess_func(
class LTX2PipelineConfig(PipelineConfig): class LTX2PipelineConfig(PipelineConfig):
"""Configuration for LTX-Video pipeline.""" """Configuration for LTX-Video pipeline."""
task_type: ModelTaskType = ModelTaskType.T2V task_type: ModelTaskType = ModelTaskType.TI2V
skip_input_image_preprocess: bool = True
dit_config: LTX2Config = field(default_factory=LTX2Config) dit_config: LTX2Config = field(default_factory=LTX2Config)
# Model architecture # Model architecture
python/sglang/multimodal_gen/runtime/pipelines_core/stages/input_validation.py
+2
View File
@@ -140,6 +140,8 @@ class InputValidationStage(PipelineStage):
batch.height = height batch.height = height
elif server_args.pipeline_config.task_type == ModelTaskType.TI2V: elif server_args.pipeline_config.task_type == ModelTaskType.TI2V:
if server_args.pipeline_config.skip_input_image_preprocess:
return
# duplicate with vae_image_processor # duplicate with vae_image_processor
# further processing for ti2v task # further processing for ti2v task
if isinstance( if isinstance(