hiyouga/LlamaFactory

Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)

What it solves

LlamaFactory simplifies the process of fine-tuning and deploying over 100 different large language models (LLMs). It removes the need for complex coding by providing a unified, user-friendly interface for training, evaluating, and serving models, making it accessible even for those without deep machine learning expertise.

How it works

The project provides a modular framework that supports a wide range of training methods—from standard supervised fine-tuning to advanced reinforcement learning techniques like DPO and PPO. Users can interact with the system through a zero-code Web UI (LLaMA Board) or a command-line interface (CLI). It integrates with popular hardware acceleration tools and supports various model architectures, allowing users to train models locally or in the cloud with minimal configuration.

Who it’s for

It is designed for developers, researchers, and AI practitioners who need to customize large language models for specific tasks—such as tool use, multi-turn dialogue, or image understanding—without building complex training pipelines from scratch.

Highlights

  • Extensive Model Support: Compatible with over 100 models, including LLaMA, Mistral, Qwen, DeepSeek, and Gemma.
  • Zero-Code Interface: Includes a Gradio-powered Web UI for visual model management and training.
  • Efficient Training: Supports advanced memory-saving techniques like QLoRA, FlashAttention-2, and Unsloth to train large models on consumer-grade hardware.
  • Deployment Ready: Offers built-in support for OpenAI-style API deployment using vLLM or SGLang for high-speed inference.
  • Experiment Tracking: Integrates with tools like TensorBoard, Wandb, and SwanLab to monitor training progress in real-time.

Related

  • Project
  • Project
  • Project
  • Project
  • Project