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

This commit is contained in:
GMI Xiao Jin
2026-02-25 01:30:36 -08:00
committed by GitHub
parent 671b595570
commit c4ef33862b
3 changed files with 5 additions and 1 deletions

View File

@@ -153,6 +153,7 @@ class PipelineConfig:
"""The base configuration class for a generation pipeline."""
task_type: ModelTaskType = ModelTaskType.I2I
skip_input_image_preprocess: bool = False
model_path: str = ""
pipeline_config_path: str | None = None

View File

@@ -115,7 +115,8 @@ def _gemma_postprocess_func(
class LTX2PipelineConfig(PipelineConfig):
"""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)
# Model architecture

View File

@@ -140,6 +140,8 @@ class InputValidationStage(PipelineStage):
batch.height = height
elif server_args.pipeline_config.task_type == ModelTaskType.TI2V:
if server_args.pipeline_config.skip_input_image_preprocess:
return
# duplicate with vae_image_processor
# further processing for ti2v task
if isinstance(