xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-13 06:04:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Core][Multimodal] Convert PIL Image to array without data copy when hashing (#18682)

Browse Source 
Signed-off-by: Lukas Geiger <lukas.geiger94@gmail.com>
This commit is contained in:
Lukas Geiger 2025-05-25 18:33:35 +01:00 committed by GitHub
parent 57fd13a707
commit 6071e989df
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/multimodal/hasher.py
View File

@ -36,8 +36,8 @@ class MultiModalHasher:
return np.array(obj).tobytes() return np.array(obj).tobytes()
if isinstance(obj, Image.Image): if isinstance(obj, Image.Image):
return cls.item_to_bytes("image", return cls.item_to_bytes(
np.array(convert_image_mode(obj, "RGBA"))) "image", np.asarray(convert_image_mode(obj, "RGBA")))
if isinstance(obj, torch.Tensor): if isinstance(obj, torch.Tensor):
return cls.item_to_bytes("tensor", obj.numpy()) return cls.item_to_bytes("tensor", obj.numpy())
if isinstance(obj, np.ndarray): if isinstance(obj, np.ndarray):

2
vllm/multimodal/video.py
View File

@ -164,7 +164,7 @@ class VideoMediaIO(MediaIO[npt.NDArray]):
) )
return np.stack([ return np.stack([
np.array(load_frame(frame_data)) np.asarray(load_frame(frame_data))
for frame_data in data.split(",") for frame_data in data.split(",")
]) ])