What problem does it solve?
Axolotl simplifies the complex process of fine-tuning Large Language Models by providing a unified, configuration-driven approach that handles diverse training methods and model architectures without requiring deep code modifications.
Core Features & Use Cases
- Unified Configuration: Manage LoRA, QLoRA, DPO, and GRPO training runs using simple YAML files.
- Broad Model Support: Access support for over 100 models and various dataset formats including instruction, chat, and pre-training.
- Use Case: A researcher needs to fine-tune a Llama-3 model on a custom dataset using QLoRA. Instead of writing custom training loops, they define the model path, dataset, and LoRA parameters in a YAML file, and Axolotl handles the tokenization, training, and checkpointing.
Quick Start
Use the axolotl skill to launch a training run using the configuration file named train_config.yaml.