Add tool and update model config

scikkk · Jul 15, 2024 · 54f1bb1 · 54f1bb1
1 parent 50998a9
commit 54f1bb1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/vlmeval/config.py b/vlmeval/config.py
@@ -80,7 +80,7 @@
 
 qwen_series = {
     'qwen_base': partial(QwenVL, model_path='Qwen/Qwen-VL'),
-    'qwen_chat': partial(QwenVL, model_path='Qwen/Qwen-VL-Chat'),
+    'qwen_chat': partial(QwenVLChat, model_path='Qwen/Qwen-VL-Chat'),
     'monkey': partial(Monkey, model_path='echo840/Monkey'),
     'monkey-chat': partial(MonkeyChat, model_path='echo840/Monkey-Chat')
 }

diff --git a/vlmeval/smp/file.py b/vlmeval/smp/file.py
@@ -243,3 +243,10 @@ def file_size(f, unit='GB'):
         'KB': 2 ** 10,
     }
     return stats.st_size / div_map[unit]
+
+
+def parquet_to_tsv(file_path):
+    data = pd.read_parquet(file_path)
+    pth = '/'.join(file_path.split('/')[:-1])
+    data_name = file_path.split('/')[-1].split('.')[0]
+    data.to_csv(osp.join(pth, f'{data_name}.tsv'), sep='\t', index=False)