Skip to content

Commit a5f9632

Browse files
authored
[Model]: Support Valley_7B (#689)
* feat: support valley eval * feat: modify the model loading method * style: fix pre-commit
1 parent aa9f50e commit a5f9632

24 files changed

+5290
-1
lines changed

vlmeval/config.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -388,6 +388,10 @@
388388
'h2ovl-mississippi-1b': partial(H2OVLChat, model_path='h2oai/h2ovl-mississippi-800m'),
389389
}
390390

391+
valley_series = {
392+
'valley_eagle': partial(ValleyEagleChat, model_path='bytedance-research/Valley-Eagle-7B'),
393+
}
394+
391395
supported_VLM = {}
392396

393397
model_groups = [
@@ -399,7 +403,7 @@
399403
mantis_series, mmalaya_series, phi3_series, xgen_mm_series, qwen2vl_series,
400404
slime_series, eagle_series, moondream_series, llama_series, molmo_series,
401405
kosmos_series, points_series, nvlm_series, vintern_series, h2ovl_series, aria_series,
402-
smolvlm_series, sail_series
406+
smolvlm_series, sail_series, valley_series
403407
]
404408

405409
for grp in model_groups:

vlmeval/vlm/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -60,3 +60,4 @@
6060
from .falcon_vlm import Falcon2VLM
6161
from .smolvlm import SmolVLM
6262
from .sail_vl import SailVL
63+
from .valley import ValleyEagleChat

vlmeval/vlm/valley/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
from .valley_eagle_chat import ValleyEagleChat
+36
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
accelerate==0.34.2
2+
bert-score==0.3.13
3+
byted-wandb==0.13.72
4+
datasets==2.21.0
5+
decord==0.6.0
6+
deepspeed==0.9.5
7+
einops==0.8.0
8+
evaluate==0.4.3
9+
fastapi==0.115.0
10+
flash_attn
11+
ftfy==6.2.3
12+
markdown2==2.5.0
13+
ninja==1.11.1.1
14+
nltk==3.9.1
15+
numpy==1.26.4
16+
omegaconf==2.3.0
17+
openai==0.28
18+
opencv-python-headless==4.10.0.84
19+
packaging==24.1
20+
pandas==2.2.2
21+
peft==0.5.0
22+
prettytable==3.11.0
23+
protobuf==3.20.3
24+
pyarrow==15.0.0
25+
pydantic==1.10.14
26+
qwen_vl_utils
27+
requests==2.32.3
28+
rouge-score==0.1.2
29+
scikit-image==0.24.0
30+
scikit-learn==1.5.2
31+
sentencepiece==0.1.97
32+
timm==0.6.7
33+
tokenizers>=0.13.3
34+
torchmetrics
35+
transformers==4.45.2
36+
uvicorn==0.30.6
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
CONTROLLER_HEART_BEAT_EXPIRATION = 30
2+
WORKER_HEART_BEAT_INTERVAL = 15
3+
CONV_SEP = "###"
4+
LOGDIR = "./valley/serve/serve_logs"
5+
SERVE_IMAGE = "./valley/serve/serve_logs"
6+
SHELL_UI_HEADER = '''
7+
8+
██╗ ██╗ █████╗ ██╗ ██╗ ███████╗██╗ ██╗ ██████╗██╗ ██╗ █████╗ ████████╗
9+
██║ ██║██╔══██╗██║ ██║ ██╔════╝╚██╗ ██╔╝ ██╔════╝██║ ██║██╔══██╗╚══██╔══╝
10+
██║ ██║███████║██║ ██║ █████╗ ╚████╔╝ ██║ ███████║███████║ ██║
11+
╚██╗ ██╔╝██╔══██║██║ ██║ ██╔══╝ ╚██╔╝ ██║ ██╔══██║██╔══██║ ██║
12+
╚████╔╝ ██║ ██║███████╗███████╗███████╗ ██║ ╚██████╗██║ ██║██║ ██║ ██║
13+
╚═══╝ ╚═╝ ╚═╝╚══════╝╚══════╝╚══════╝ ╚═╝ ╚═════╝╚═╝ ╚═╝╚═╝ ╚═╝ ╚═╝
14+
15+
'''

0 commit comments

Comments
 (0)