导航菜单
切换主题
框架库/Axolotl
Axolotl

Axolotl

训练框架开源
8.5k Stars·Apache-2.0

Axolotl 是一个专注于大语言模型微调的开源工具,提供声明式 YAML 配置,支持多种模型和训练方法。

框架介绍

Axolotl 是一个开源的大语言模型微调工具,旨在简化微调流程并提高可复现性。它采用声明式 YAML 配置方式,用户只需定义配置文件即可完成训练,无需编写复杂代码。 Axolotl 的核心优势在于其对训练方法的全面支持——包括 LoRA、QLoRA、Full Fine-tuning 等主流方法,同时支持多 GPU 分布式训练和 DeepSpeed 集成。它与 Hugging Face 生态深度集成,支持大部分主流开源模型。

核心特性

1

YAML 配置驱动

声明式配置文件定义训练,简化流程并提高可复现性

2

多训练方法

支持 LoRA、QLoRA、Full Fine-tuning、Freeze 等训练策略

3

分布式训练

原生支持多 GPU 训练,集成 DeepSpeed 和 FSDP

4

Hugging Face 集成

与 Hugging Face 生态无缝集成,支持大部分开源模型

5

数据格式支持

支持 alpaca、sharegpt、jsonl 等多种数据格式

6

自动实验追踪

集成 wandb、mlflow 等实验追踪工具

应用场景

模型微调

使用 LoRA/QLoRA 高效微调大语言模型

实验管理

通过配置文件管理多个训练实验

多 GPU 训练

在多 GPU 环境下进行分布式微调

模型实验

快速尝试不同模型和训练配置

适用人群与场景

ML 工程师

进行大模型微调和实验的工程师

AI 研究员

需要复现和管理训练实验的研究人员

开源贡献者

参与开源模型训练的社区成员

DevOps 工程师

构建自动化训练流程的运维人员

README