xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-01-02 19:39:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Avoid unnecessary HF do_rescale warning when passing dummy data (#15107)

Browse Source 
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung 2025-03-19 18:42:31 +08:00 committed by GitHub
parent dafb4e504a
commit 1fe0fd12d3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/multimodal/profiling.py
View File

@ -73,7 +73,7 @@ class BaseDummyInputsBuilder(ABC, Generic[_I]):
height: int,
num_images: int,
) -> list[Image.Image]:
image = Image.new("RGB", (width, height), color=0)
image = Image.new("RGB", (width, height), color=255)
return [image] * num_images
def _get_dummy_videos(
@ -84,7 +84,7 @@ class BaseDummyInputsBuilder(ABC, Generic[_I]):
num_frames: int,
num_videos: int,
) -> list[npt.NDArray]:
video = np.zeros((num_frames, width, height, 3))
video = np.full((num_frames, width, height, 3), 255)
return [video] * num_videos