Rename InternVL models

OpenGVLab · Apr 26, 2024 · f067331 · f067331
1 parent 8ec8203
commit f067331
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/vlmeval/config.py b/vlmeval/config.py
@@ -68,10 +68,10 @@
 }
 
 internvl_series = {
-    'InternVL-Chat-V1-1':partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-Chinese-V1-1'),
-    'InternVL-Chat-V1-2': partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-Chinese-V1-2'),
-    'InternVL-Chat-V1-2-Plus': partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-Chinese-V1-2-Plus'),
-    'InternVL-Chat-V1-5': partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-V1-5-Preview'),
+    'InternVL-Chat-V1-1':partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-V1-1'),
+    'InternVL-Chat-V1-2': partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-V1-2'),
+    'InternVL-Chat-V1-2-Plus': partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-V1-2-Plus'),
+    'InternVL-Chat-V1-5': partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-V1-5'),
 }
 
 yivl_series = {
@@ -134,4 +134,4 @@
     import sys
     ver = sys.argv[1]
     if ver in transformer_ver:
-        print(' '.join(transformer_ver[ver]))
+        print(' '.join(transformer_ver[ver]))