Skip to content

Commit 44d698f

Browse files
authored
[Model] Add support for varco-vision-2.0-1.7b (#1184)
* add 1.7b model * fix vqa prompt as some of them doesn't work * minor fix * Revert "minor fix" This reverts commit f44f231.
1 parent 76d6258 commit 44d698f

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

vlmeval/config.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -726,6 +726,9 @@
726726
"varco-vision-hf": partial(
727727
LLaVA_OneVision_HF, model_path="NCSOFT/VARCO-VISION-14B-HF"
728728
),
729+
"varco-vision-2-1.7b": partial(
730+
VarcoVision, model_path="NCSOFT/VARCO-VISION-2.0-1.7B"
731+
),
729732
"varco-vision-2-14b": partial(
730733
VarcoVision, model_path="NCSOFT/VARCO-VISION-2.0-14B"
731734
),

vlmeval/vlm/varco_vision.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,12 +70,13 @@ def set_res(self, dataset):
7070
self.set_grid(8)
7171

7272
def use_custom_prompt(self, dataset):
73+
if any(dataset.startswith(prefix) for prefix in
74+
['MMVet', 'MathVista', 'MathVerse', 'MathVision', 'LLaVABench']):
75+
return True
7376
if DATASET_TYPE(dataset) == 'Y/N':
7477
return True
7578
if DATASET_TYPE(dataset) == 'MCQ':
7679
return True
77-
if DATASET_TYPE(dataset) == 'VQA' and not dataset.startswith('OCRBench'):
78-
return True
7980
return False
8081

8182
def build_prompt(self, line, dataset=None):

0 commit comments

Comments
 (0)