From 6b6fcbba726c2fad9c19b03ea58d8bba02e149be Mon Sep 17 00:00:00 2001 From: Yuliang Liu <34134635+Yuliang-Liu@users.noreply.github.com> Date: Sat, 24 Aug 2024 10:39:31 +0800 Subject: [PATCH] Update MiniMonkey.py --- scripts/MiniMonkey.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/MiniMonkey.py b/scripts/MiniMonkey.py index d9cc098..c697557 100644 --- a/scripts/MiniMonkey.py +++ b/scripts/MiniMonkey.py @@ -195,9 +195,9 @@ def eval_worker(args, data, eval_id, output_queue): image_path = os.path.join(args.image_folder, data[i]['image_path']) qs = data[i]['question'] - pixel_values, target_aspect_ratio = load_image(image_path, min_num=9, max_num=24) + pixel_values, target_aspect_ratio = load_image(image_path, min_num=12, max_num=24) pixel_values = pixel_values.to(f'cuda:{eval_id}').to(torch.bfloat16) - pixel_values2 = load_image2(image_path, target_aspect_ratio=target_aspect_ratio, min_num=5, max_num=8) + pixel_values2 = load_image2(image_path, target_aspect_ratio=target_aspect_ratio, min_num=3, max_num=11) pixel_values2 = pixel_values2.to(f'cuda:{eval_id}').to(torch.bfloat16) pixel_values = torch.cat((pixel_values[:-1], pixel_values2[:-1], pixel_values[-1:]), 0)