2U1/Qwen-VL-Series-Finetune
An open-source implementaion for fine-tuning Qwen-VL series by Alibaba Cloud.
解决的问题
该项目提供了一个全面的工具包,用于微调Qwen-VL系列视觉语言模型,包括Qwen2-VL、Qwen2.5-VL、Qwen3-VL和Qwen3.5。它简化了将这些模型适配到特定任务的过程,支持从全量微调到LoRA和QLoRA等参数高效技术的多种训练方法。
工作原理
该仓库包含利用HuggingFace和Liger-Kernel进行内存高效训练的脚本。它支持多种训练范式,包括监督微调(SFT)、直接偏好优化(DPO)和组相对策略优化(GRPO)。该系统处理混合模态数据集(图像和视频),并为基于推理的模型提供特定格式,以确保训练与模型的推理时聊天模板匹配。
适用人群
专为希望针对特定领域的视觉理解、视频分析或推理任务定制Qwen-VL模型的AI研究人员和开发人员设计。
亮点
- 多样化的训练方法:支持全量微调、LoRA、QLoRA、DPO和GRPO。
- 多模态支持:能够在单图像、多图像和视频数据集上进行训练。
- 推理优化:针对Qwen3-VL-Thinking和Qwen3.5模型的推理轨迹训练提供专门支持。
- 效率工具:集成Liger-Kernel和DeepSpeed(Zero2/Zero3)以减少内存使用并提高速度。
- 灵活的冻结选项:可独立冻结LLM、视觉塔或投影器,或仅解冻顶部k层。
- 分类支持:包含使用可选MLP层进行分类微调的脚本。
相关
- 项目
- Dispatch
- 项目
- Dispatch
- Dispatch