用于合并预训练大语言模型的工具包,支持多种合并算法,可在 CPU 或 GPU 上运行。
TinyLLaVA/TinyLLaVA_Factory
基于 PyTorch 与 Hugging Face 的模块化代码库,用于构建、训练和评估小规模多模态视觉语言大模型。
- Star
- 1,011
- Fork
- 100
- 未关闭 Issue
- 114
- 最近提交
- 19天前
这是什么?
TinyLLaVA Factory 是一个面向小规模多模态大模型的模块化代码库,将视觉编码器与语言模型组合在一起。LLM、视觉塔、连接器和训练方案可以相互独立替换,便于在不重写训练代码的前提下定制自己的视觉语言模型。
框架支持 OpenELM、TinyLlama、StableLM、Qwen、Gemma、Phi 等语言模型,视觉部分支持 CLIP、SigLIP、DINO 及两者组合,连接器支持 MLP、Qformer 和 Resampler。训练方式包括冻结、全量与部分微调,以及 LoRA、QLoRA。项目还发布了多个预训练模型,如 TinyLLaVA-Phi-2-SigLIP-3.1B,并给出 VQA-v2、GQA、TextVQA、MME 等基准结果。
仓库提供 Gradio 网页演示、命令行推理工具、训练脚本示例,以及数据准备和模型评测的文档。
核心功能
- 01LLM、视觉塔和连接器组件可自由替换,用于构建定制的多模态模型。
- 02支持冻结、全量、部分微调,以及 LoRA 和 QLoRA 训练方案。
- 03提供基于 Phi-2、Gemma、Qwen、TinyLlama 的预训练模型,视觉编码器为 SigLIP 或 CLIP。
- 04提供 Gradio 网页演示和命令行推理工具。
- 05附带数据准备与模型评测的说明文档。
适用场景
- 在单张 GPU 上训练小型视觉语言模型,用于图像问答。
- 在模块化框架中对比不同语言模型与视觉编码器的效果。
- 使用 LoRA 在自定义图文数据集上微调多模态模型。
- 复现小规模多模态大模型的训练与评测结果。
快速上手
- 1
克隆仓库
git clone https://github.com/TinyLLaVA/TinyLLaVA_Factory.git - 2
进入项目文件夹
cd TinyLLaVA_Factory - 3
创建并激活 conda 环境,然后安装本包
conda create -n tinyllava_factory python=3.10 -y && conda activate tinyllava_factory && pip install --upgrade pip && pip install -e . - 4
使用预训练模型启动本地 Gradio 网页演示
python tinyllava/serve/app.py --model-path tinyllava/TinyLLaVA-Phi-2-SigLIP-3.1B
使用前须知
- 训练及大多数推理需要 conda 环境和 CUDA GPU,且 flash-attn 需要单独编译安装。
- 其环境要求与 LLaVA 不同,已有的 LLaVA 环境可能无法直接复用。
收录于专题
标签
常见问题
- TinyLLaVA_Factory 是什么?
- 基于 PyTorch 与 Hugging Face 的模块化代码库,用于构建、训练和评估小规模多模态视觉语言大模型。 TinyLLaVA Factory 是一个面向小规模多模态大模型的模块化代码库,将视觉编码器与语言模型组合在一起。LLM、视觉塔、连接器和训练方案可以相互独立替换,便于在不重写训练代码的前提下定制自己的视觉语言模型。
- TinyLLaVA_Factory 可以免费使用吗?用的什么许可证?
- TinyLLaVA_Factory 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
- TinyLLaVA_Factory 还在维护吗?
- 活跃维护中。TinyLLaVA_Factory 最近一次提交是在 2026年10月19日。
- TinyLLaVA_Factory 适合用来做什么?
- 在单张 GPU 上训练小型视觉语言模型,用于图像问答。 在模块化框架中对比不同语言模型与视觉编码器的效果。 使用 LoRA 在自定义图文数据集上微调多模态模型。 复现小规模多模态大模型的训练与评测结果。
- TinyLLaVA_Factory 怎么安装和运行?
- 克隆仓库: git clone https://github.com/TinyLLaVA/TinyLLaVA_Factory.git 进入项目文件夹: cd TinyLLaVA_Factory 创建并激活 conda 环境,然后安装本包: conda create -n tinyllava_factory python=3.10 -y && conda activate tinyllava_factory && pip install --upgrade pip && pip install -e . 使用预训练模型启动本地 Gradio 网页演示: python tinyllava/serve/app.py --model-path tinyllava/TinyLLaVA-Phi-2-SigLIP-3.1B
- 在哪里获取 TinyLLaVA_Factory?
- 源代码和安装说明在 GitHub 上:https://github.com/TinyLLaVA/TinyLLaVA_Factory
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
从头预训练并指令微调小参数量中文 LLaMA2 模型的完整代码,单张 24G 显卡即可运行。
一份精选合集,汇集大语言模型微调相关的教程、论文、工具与课程资源。
Hugging Face 参数高效微调(PEFT)库,只训练少量额外参数即可适配大型预训练模型。
Hugging Face 推出的库,用 SFT、GRPO、DPO、KTO 等方法对 Transformer 语言模型做后训练。
将 BERT、RoBERTa、XLNet 等 Hugging Face 预训练 Transformer 模型接入 spaCy 流程的组件与架构。