From d3a13425d94f241603ae1810bc01b43622c98984 Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Mon, 19 Jan 2026 04:34:38 +0000 Subject: [PATCH] Fix DiffusionOutput return statement in LTX2Pipeline Co-authored-by: hsliuustc0106 <222337142+hsliuustc0106@users.noreply.github.com> --- vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py b/vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py index ba95db235..4cc2de28b 100644 --- a/vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py +++ b/vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py @@ -190,7 +190,4 @@ class LTX2Pipeline(nn.Module): frames = output # Return as DiffusionOutput - return DiffusionOutput( - images=frames, - final_output_type="image", - ) + return DiffusionOutput(output=frames)