diff --git a/vllm/benchmarks/datasets.py b/vllm/benchmarks/datasets.py index b1aa8530eb02..1704130d9131 100644 --- a/vllm/benchmarks/datasets.py +++ b/vllm/benchmarks/datasets.py @@ -30,7 +30,6 @@ from io import BytesIO from tempfile import NamedTemporaryFile from typing import Any, cast -import cv2 import numpy as np from PIL import Image from transformers import PreTrainedTokenizerBase @@ -850,6 +849,8 @@ class RandomMultiModalDataset(RandomDataset): Creates a video with random pixel values, encodes it to MP4 format, and returns the content as bytes. """ + import cv2 + random_pixels = self._rng.integers( 0, 256,