From e542caea3cf5d362131bccdb97a70ce50839c332 Mon Sep 17 00:00:00 2001 From: kennymckormick Date: Tue, 24 Sep 2024 17:25:32 +0800 Subject: [PATCH] [Minor] Set the kwargs default for internvl_chat. Max_new_tokens to 1024. --- vlmeval/vlm/internvl_chat.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/vlmeval/vlm/internvl_chat.py b/vlmeval/vlm/internvl_chat.py index 89cda430..16bee8f8 100644 --- a/vlmeval/vlm/internvl_chat.py +++ b/vlmeval/vlm/internvl_chat.py @@ -174,7 +174,10 @@ def __init__(self, model_path='OpenGVLab/InternVL-Chat-V1-5', load_in_8bit=False self.image_size = self.model.config.vision_config.image_size self.version = version - self.kwargs = kwargs + kwargs_default = dict(do_sample=False, max_new_tokens=1024, top_p=None, num_beams=1) + kwargs_default.update(kwargs) + self.kwargs = kwargs_default + warnings.warn(f'Following kwargs received: {self.kwargs}, will use as generation config. ') def use_custom_prompt(self, dataset):