From 5d19005525948184270cc1558ae42ed2f26bc90c Mon Sep 17 00:00:00 2001 From: kennymckormick Date: Sun, 7 Apr 2024 15:54:02 +0800 Subject: [PATCH] update --- vlmeval/config.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/vlmeval/config.py b/vlmeval/config.py index 51170120d..48641c165 100644 --- a/vlmeval/config.py +++ b/vlmeval/config.py @@ -121,3 +121,18 @@ for grp in model_groups: supported_VLM.update(grp) + +transformer_ver = {} +transformer_ver['4.33.0'] = list(qwen_series) + list(internvl_series) + list(xcomposer_series) + [ + 'mPLUG-Owl2', 'flamingov2', 'VisualGLM_6b', 'MMAlaya', 'PandaGPT_13B' +] + list(idefics_series) + list(minigpt4_series) + list(instructblip_series) +transformer_ver['4.37.0'] = [x for x in llava_series if 'next' not in x] + [ + 'TransCore_M', 'cogvlm-chat', 'cogvlm-grounding-generalist', 'emu2', 'emu2_chat', 'MiniCPM-V', 'OmniLMM_12B' +] + list(xtuner_series) + list(yivl_series) + list(deepseekvl_series) +transformer_ver['4.39.0'] = [x for x in llava_series if 'next' in x] + +if __name__ == '__main__': + import sys + ver = sys.argv[1] + if ver in transformer_ver: + print(' '.join(transformer_ver[ver])) \ No newline at end of file