[BugFix] Fix the issue with stream=True (#2955)

Signed-off-by: amy-why-3459 <wuhaiyan17@huawei.com>
This commit is contained in:
amy-why-3459 authored and GitHub committed 2026-04-21 11:51:39 +08:00
1 parent de4e47237a
commit c1ba86ab5b
1 file changed
+3 -4
+3 -4
View File
@@ -118,10 +118,9 @@ class OmniRequestState(RequestState):
if isinstance(v, list) and v and isinstance(v[0], torch.Tensor):
try:
if k == "audio":
# Concatenate delta audio chunks (1-D) into the full waveform.
# Each entry is a per-step slice; flatten to -1 so chunks with
# inconsistent leading dims can still be joined on the sample axis.
self.mm_accumulated[k] = torch.cat([t.reshape(-1) for t in v], dim=0)
# When the audio tensor shape is inconsistent, torch.cat will fail.
# We need to use torch.cat in -1 dimension.
continue
elif k == "sr":
# Sample rate is a constant scalar, keep last value.
self.mm_accumulated[k] = v[-1]