anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[Performance] diffusion timing (#1757)

Browse Source 
Signed-off-by: dengyunyang <584797741@qq.com>
Signed-off-by: bjf-frz <frz123db@gmail.com>
Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com>
Co-authored-by: bjf-frz <frz123db@gmail.com>
Co-authored-by: Didan Deng <33117903+wtomin@users.noreply.github.com>
This commit is contained in:
dengyunyangbjf-frzDidan Deng
 authored and GitHub committed 2026-03-19 12:26:12 +08:00
1 parent ae92ef5f00
commit fde4808723
43 files changed
+496 -38
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vllm_omni/diffusion/diffusion_engine.py
+4
View File
@@ -170,6 +170,7 @@ class DiffusionEngine:
latents=output.trajectory_latents,
multimodal_output={"audio": request_audio_payload},
final_output_type="audio",
stage_durations=output.stage_durations,
),
]
else:
@@ -185,6 +186,7 @@ class DiffusionEngine:
latents=output.trajectory_latents,
custom_output=output.custom_output or {},
multimodal_output=mm_output,
stage_durations=output.stage_durations,
),
]
else:
@@ -214,6 +216,7 @@ class DiffusionEngine:
latents=output.trajectory_latents,
multimodal_output={"audio": request_audio_payload},
final_output_type="audio",
stage_durations=output.stage_durations,
),
)
else:
@@ -239,6 +242,7 @@ class DiffusionEngine:
latents=output.trajectory_latents,
custom_output=output.custom_output or {},
multimodal_output=mm_output,
stage_durations=output.stage_durations,
),
)