xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-11 13:55:01 +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()
if isinstance(obj, Image.Image):
return cls.item_to_bytes("image",
np.array(convert_image_mode(obj, "RGBA")))
return cls.item_to_bytes(
"image", np.asarray(convert_image_mode(obj, "RGBA")))
if isinstance(obj, torch.Tensor):
return cls.item_to_bytes("tensor", obj.numpy())
if isinstance(obj, np.ndarray):

2
vllm/multimodal/video.py
View File

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