安装准备
首先需要配置Python 3.9环境,建议使用conda管理虚拟环境。
安装步骤
- 克隆仓库:在终端运行
git clone https://github.com/OpenGVLab/InternVL.git
并进入目录 - 创建虚拟环境:使用
conda create -n internvl python=3.9 -y
创建环境 - 安装基础依赖:运行
pip install -r requirements.txt
安装核心库
可选安装
- 安装Flash-Attention以加速推理:
pip install flash-attn==2.3.6
- 安装MMDeploy用于生产部署:
mim install mmdeploy
多模态对话使用
下载模型(如InternVL2_5-8B)后,可以使用以下代码进行多模态对话:
from lmdeploy import pipeline
from lmdeploy.vl import load_image
model = 'OpenGVLab/InternVL2_5-8B'
image = load_image('tiger.jpeg')
pipe = pipeline(model)
response = pipe(('描述这张图片', image))
print(response.text)
注意事项
8B模型约需16GB GPU内存,处理高分辨率图像时可能需要更多资源。
本答案来源于文章《InternVL:开源多模态大模型,支持图像、视频和文本处理》