From 1dae53731389cce1d2f99697c920a4c159196548 Mon Sep 17 00:00:00 2001
From: Mu GuanLin <1203789601@qq.com>
Date: Thu, 3 Sep 2026 18:48:43 +0800
Subject: [PATCH] [Bugfix][LTX] Restore distilled two-stage video serving
 (#6847)

Signed-off-by: mglyn <1203789601@qq.com>
---
 tests/entrypoints/openai_api/test_video_pipeline_capability.py | 1 +
 vllm_omni/diffusion/model_metadata.py                          | 1 +
 2 files changed, 2 insertions(+)

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",
 }