InternVL 部署微调实践

参考文档:Tutorial/docs/L2/InternVL at camp4 · InternLM/Tutorialhttps://github.com/InternLM/Tutorial/tree/camp4/docs/L2/InternVL

1.环境配置

1.1.训练环境配置

新建虚拟环境并进入:

conda create --name xtuner-env python=3.10 -y
conda activate xtuner-env

"xtuner-env"为训练环境名,可以根据个人喜好设置,在本教程中后续提到训练环境均指"xtuner-env"环境。我这里是newxtuner

安装与deepspeed集成的xtuner和相关包:

pip install xtuner==0.1.23 timm==1.0.9
pip install 'xtuner[deepspeed]'
pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu121
pip install transformers==4.39.0 tokenizers==0.15.2 peft==0.13.2 datasets==3.1.0 accelerate==1.2.0 huggingface-hub==0.26.5 

1.2.推理环境配置

配置推理所需环境:

conda create -n lmdeploy python=3.10 -y
conda activate lmdeploy
pip install lmdeploy==0.6.1 gradio==4.44.1 timm==1.0.9

"lmdeploy"为推理使用环境名。

2.LMDeploy部署

2.1.LMDeploy基本用法介绍

我们主要通过pipeline.chat 接口来构造

### InternVL2 部署指南 #### 1. 数据准备 为了训练和微调模型,官方提供了 `InternVL-Chat-V1-2-SFT-Data` 数据集。该数据集包含了用于监督微调的高质量对话样本[^1]。确保下载并解压此数据集到指定路径。 #### 2. 环境搭建 环境配置是成功部署的关键部分之一。以下是基于文档中的指导完成环境设置的方法: - **克隆项目仓库**: 使用以下命令获取最新版本的 InternVL 源码。 ```bash git clone https://gitcode.com/gh_mirrors/in/InternVL.git ``` - **依赖安装**: 进入项目目录后执行如下脚本以安装必要的 Python 库和其他工具链。 ```bash cd InternVL && pip install -r requirements.txt ``` 以上步骤均来自项目的具体说明文档[^3]。 #### 3. 修改 XTuner 的 Config 文件 针对特定任务需求调整超参数以及网络结构定义非常重要。对于 InternVL2 而言,在 `/root/InternLM/code/XTuner/xtuner/configs/internvl/v2/internvl_v2_internlm2_2b_qlora_finetune.py` 中有若干可调节项需注意更改[^4]。例如学习率、批次大小等都可能影响最终效果表现。 #### 4. 实践量化部署 通过 LMDeploy Lite 工具包实现高效资源利用下的性能优化成为当前主流趋势之一。选取 InternVL2-26B 版本不仅因为其具备处理图文混合输入的能力,更重要的是借此机会深入理解整个流程从而掌握更高级别的技术应用能力[^2]。 ```python from lmdeploy.lite import get_model_info, load_from_file model_path = 'path/to/model' info = get_model_info(model_path) print(info) # 加载模型实例化对象 session = load_from_file(model_path=model_path) output = session('your prompt here') ``` 上述代码片段展示了如何加载预训练好的权重文件并通过会话接口发起请求得到响应结果。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值