in eval/benchmark_suite.py [0:0]
def benchmark_vlm(
vit_model_type: str,
lm_model_type: str,
lm_tokenizer_path: str,
mp_pixel_shuffle_factor: int,
image_path: str,
prompt: str,
max_new_tokens: int,
num_runs: int,
warmup_runs: int,
device: torch.device,