[BugFix] Fix NoneType' object has no attribute 'detach' (#2797)

Signed-off-by: amy-why-3459 <wuhaiyan17@huawei.com>
This commit is contained in:
amy-why-3459 authored and GitHub committed 2026-04-14 10:26:55 -04:00
1 parent 53a9cf49a6
commit f03ab38783
2 files changed
+6 -7

No files matched your search

+1 -1
View File
@@ -120,7 +120,7 @@ def test_mix_to_text_audio_001(omni_server, openai_client) -> None:
}
# Test single completion
openai_client.send_omni_request(request_config)
openai_client.send_omni_request(request_config, request_num=get_max_batch_size())
@pytest.mark.advanced_model
+5 -6
View File
@@ -797,12 +797,11 @@ class GPUARModelRunner(OmniGPUModelRunner, OmniConnectorModelRunnerMixin):
elif isinstance(v, dict):
mm_payload[k] = {sk: sv[start:end].contiguous() for sk, sv in v.items()}
elif isinstance(v, list):
if idx < len(v):
element = v[idx]
if element is not None:
if isinstance(element, torch.Tensor):
element = element.clone()
mm_payload[k] = element
element = v[idx] if idx < len(v) else v[0]
if element is not None:
if isinstance(element, torch.Tensor):
element = element.clone()
mm_payload[k] = element
# Skip None elements: msgspec cannot serialize None
# in dict[str, torch.Tensor] typed fields.
elif isinstance(v, torch.Tensor):