Skip to content

Commit 1fe0fd1

Browse files
[Misc] Avoid unnecessary HF do_rescale warning when passing dummy data (vllm-project#15107)
Signed-off-by: DarkLight1337 <[email protected]>
1 parent dafb4e5 commit 1fe0fd1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/multimodal/profiling.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def _get_dummy_images(
7373
height: int,
7474
num_images: int,
7575
) -> list[Image.Image]:
76-
image = Image.new("RGB", (width, height), color=0)
76+
image = Image.new("RGB", (width, height), color=255)
7777
return [image] * num_images
7878

7979
def _get_dummy_videos(
@@ -84,7 +84,7 @@ def _get_dummy_videos(
8484
num_frames: int,
8585
num_videos: int,
8686
) -> list[npt.NDArray]:
87-
video = np.zeros((num_frames, width, height, 3))
87+
video = np.full((num_frames, width, height, 3), 255)
8888
return [video] * num_videos
8989

9090

0 commit comments

Comments
 (0)