diff --git a/tests/entrypoints/openai_api/test_video_pipeline_capability.py b/tests/entrypoints/openai_api/test_video_pipeline_capability.py index 985aadf89..530c6e216 100644 --- a/tests/entrypoints/openai_api/test_video_pipeline_capability.py +++ b/tests/entrypoints/openai_api/test_video_pipeline_capability.py @@ -57,6 +57,7 @@ def test_video_pipeline_rejects_non_video_final_outputs(stage_configs): [ "LTX2TwoStagePipeline", "LTX2DistilledOneStagePipeline", + "LTX2DistilledTwoStagePipeline", "WanDMDPipeline", "LingBotWorldCausalDMDPipeline", "LongCatVideoAvatarPipeline", diff --git a/vllm_omni/diffusion/model_metadata.py b/vllm_omni/diffusion/model_metadata.py index 7f15955ec..fcb4075ec 100644 --- a/vllm_omni/diffusion/model_metadata.py +++ b/vllm_omni/diffusion/model_metadata.py @@ -99,6 +99,7 @@ _DIFFUSION_MODEL_METADATA_ALIASES = { "WanDMDPipeline": "WanPipeline", "LTX2TwoStagePipeline": "LTX2Pipeline", "LTX2DistilledOneStagePipeline": "LTX2DistilledPipeline", + "LTX2DistilledTwoStagePipeline": "LTX2DistilledPipeline", "LingBotWorldCausalDMDPipeline": "LingBotVideoPipeline", }