EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix multimodal hashing error (#5174)

Browse Source
This commit is contained in:
fzyzcjy
2025-04-09 09:42:26 +08:00
committed by GitHub
parent 11d760d56a
commit 466899e69c
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/sglang/srt/managers/schedule_batch.py
View File

@@ -200,6 +200,7 @@ class MultimodalDataItem:
def tensor_hash(f): def tensor_hash(f):
f_list = flatten_nested_list(f) f_list = flatten_nested_list(f)
f_list = [x.flatten() if isinstance(x, torch.Tensor) else x for x in f_list]
f_cat = torch.concat(f_list).contiguous().numpy().tobytes() f_cat = torch.concat(f_list).contiguous().numpy().tobytes()
return hash(f_cat) return hash(f_cat)