Skip to content

GitLab

"sgl-kernel/git@developer.sourcefind.cn:change/sglang.git" did not exist on "8ab7d93c2e0da6bb0f3b08a43b57f2d604b57fe0"
Unverified Commit 466899e6 authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Fix multimodal hashing error (#5174)

parent 11d760d5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
python/sglang/srt/managers/schedule_batch.py
View file @ 466899e6
...@@ -200,6 +200,7 @@ class MultimodalDataItem: ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment