README.md 3.52 KB
Newer Older
wanglch's avatar
wanglch committed
1
# InternVL2.5
wanglch's avatar
wanglch committed
2
3
4
## 论文
[Expanding Performance Boundaries of Open-Source Multimodal Models with Model, Data, and Test-Time Scaling](https://arxiv.org/abs/2412.05271)

chenych's avatar
chenych committed
5
6
7
## 模型简介
InternVL 2.5 保留了与前代模型 InternVL 1.5 和 2.0 相同的模型架构,遵循“ViT-MLP-LLM”范式。在新版本中,使用随机初始化的 MLP 投影器集成了一个新增量预训练的 InternViT 与各种预训练 LLM(包括 InternLM 2.5 和 Qwen 2.5)。
正如之前的版本一样,应用了像素解混操作,将视觉标记的数量减少到原来的四分之一。此外,采用了与InternVL 1.5类似的动态分辨率策略,将图像分割成448×448像素的块。从InternVL 2.0开始的关键区别在于,额外引入了对多图像和视频数据的支持。
wanglch's avatar
wanglch committed
8
9

<div align=center>
10
    <img src="./doc/arch.png"/>
wanglch's avatar
wanglch committed
11
12
</div>

chenych's avatar
chenych committed
13
14
15
## 环境依赖
| 软件 | 版本 |
| :------: | :------: |
16
| DTK | 25.04.2 |
chenych's avatar
chenych committed
17
| python | 3.10 |
18
19
20
| torch | 2.5.1+das.opt1.dtk25042 |
| transformers | 4.37.2 |
| flash-attn | 2.6.1+das.opt1.dtk2504 |
wanglch's avatar
wanglch committed
21

chenych's avatar
chenych committed
22
推荐使用镜像: image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.2-py3.10
wanglch's avatar
wanglch committed
23

chenych's avatar
chenych committed
24
- 挂载地址`-v``{docker_name}``{docker_image_name}`根据实际模型情况修改
wanglch's avatar
wanglch committed
25

chenych's avatar
chenych committed
26
```bash
chenych's avatar
chenych committed
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
docker run -it \
    --shm-size 60g \
    --network=host \
    --name intern2.5-vl \
    --privileged \
    --device=/dev/kfd \
    --device=/dev/dri \
    --device=/dev/mkfd \
    --group-add video \
    --cap-add=SYS_PTRACE \
    --security-opt seccomp=unconfined \
    -u root \
    -v /opt/hyhal/:/opt/hyhal/:ro \
    -v /path/your_code_data/:/path/your_code_data/ \
    image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.2-py3.10 bash
wanglch's avatar
wanglch committed
42
```
chenych's avatar
chenych committed
43
更多镜像可前往[光源](https://sourcefind.cn/#/service-list)下载使用。
wanglch's avatar
wanglch committed
44

chenych's avatar
chenych committed
45
46
47
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.sourcefind.cn/tool/)开发者社区下载安装,其它包参照requirements.txt安装:
```bash
pip install -r requirements.txt
wanglch's avatar
wanglch committed
48
```
49
> 如果出现vllm等包提示版本不兼容,可忽略。
wanglch's avatar
wanglch committed
50
51

## 数据集
chenych's avatar
chenych committed
52
暂无
wanglch's avatar
wanglch committed
53
54

## 训练
chenych's avatar
chenych committed
55
暂无
wanglch's avatar
wanglch committed
56
57

## 推理
chenych's avatar
chenych committed
58
59
60
61
62
### transformers
#### 单机推理
此处以[OpenGVLab/InternVL2_5-26B](https://www.modelscope.cn/models/OpenGVLab/InternVL2_5-26B)为例
```bash
export HIP_VISIBLE_DEVICES=0,1
wanglch's avatar
wanglch committed
63
64
65
python internvl_inference.py
```

chenych's avatar
chenych committed
66
## 效果展示
wanglch's avatar
wanglch committed
67
68
- 多模态推理
<div align=left>
69
    <img src="./doc/result.png"/>
wanglch's avatar
wanglch committed
70
71
72
</div>

### 精度
chenych's avatar
chenych committed
73
DCU与GPU精度一致,推理框架:transformers。
wanglch's avatar
wanglch committed
74
75

## 预训练权重
chenych's avatar
chenych committed
76
77
78
79
80
81
82
83
84
| 模型名称  | 权重大小  | DCU型号  | 最低卡数需求 |下载地址|
|:-----:|:----------:|:----------:|:---------------------:|:----------:|
| InternVL 2.5 | 1B | K100AI| 1 | [Modelscope](https://www.modelscope.cn/models/OpenGVLab/InternVL2_5-1B) |
| InternVL 2.5 | 2B | K100AI| 1 | [Modelscope](https://www.modelscope.cn/models/OpenGVLab/InternVL2_5-2B) |
| InternVL 2.5 | 4B | K100AI| 1 | [Modelscope](https://www.modelscope.cn/models/OpenGVLab/InternVL2_5-4B) |
| InternVL 2.5 | 8B | K100AI| 1 | [Modelscope](https://www.modelscope.cn/models/OpenGVLab/InternVL2_5-8B) |
| InternVL 2.5 | 26B | K100AI| 2 | [Modelscope](https://www.modelscope.cn/models/OpenGVLab/InternVL2_5-26B) |
| InternVL 2.5 | 38B | K100AI| 2 | [Modelscope](https://www.modelscope.cn/models/OpenGVLab/InternVL2_5-38B) |
| InternVL 2.5 | 78B | K100AI| 4 | [Modelscope](https://www.modelscope.cn/models/OpenGVLab/InternVL2_5-78B) |
wanglch's avatar
wanglch committed
85

wanglch's avatar
wanglch committed
86
87
88
89
90
## 源码仓库及问题反馈
- https://developer.sourcefind.cn/codes/modelzoo/internvl2.5_pytorch

## 参考资料
- https://github.com/OpenGVLab/InternVL