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

[Perf] Qwen-Image Performance Nightly CI test (#1805)

Browse Source 
Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com>
This commit is contained in:
Didan Deng authored and GitHub committed 2026-03-24 00:19:20 +08:00
1 parent df93b3a965
commit 1f5eca5686
13 files changed
+1291 -4
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
@@ -180,6 +180,7 @@ class DiffusionEngine:
multimodal_output={"audio": request_audio_payload},
final_output_type="audio",
stage_durations=output.stage_durations,
peak_memory_mb=output.peak_memory_mb,
),
]
else:
@@ -196,6 +197,7 @@ class DiffusionEngine:
custom_output=output.custom_output or {},
multimodal_output=mm_output,
stage_durations=output.stage_durations,
peak_memory_mb=output.peak_memory_mb,
),
]
else:
@@ -226,6 +228,7 @@ class DiffusionEngine:
multimodal_output={"audio": request_audio_payload},
final_output_type="audio",
stage_durations=output.stage_durations,
peak_memory_mb=output.peak_memory_mb,
),
)
else:
@@ -252,6 +255,7 @@ class DiffusionEngine:
custom_output=output.custom_output or {},
multimodal_output=mm_output,
stage_durations=output.stage_durations,
peak_memory_mb=output.peak_memory_mb,
),
)