English

LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Models

Computation and Language 2024-07-01 v4 Artificial Intelligence

Abstract

Efficient fine-tuning is vital for adapting large language models (LLMs) to downstream tasks. However, it requires non-trivial efforts to implement these methods on different models. We present LlamaFactory, a unified framework that integrates a suite of cutting-edge efficient training methods. It provides a solution for flexibly customizing the fine-tuning of 100+ LLMs without the need for coding through the built-in web UI LlamaBoard. We empirically validate the efficiency and effectiveness of our framework on language modeling and text generation tasks. It has been released at https://github.com/hiyouga/LLaMA-Factory and received over 25,000 stars and 3,000 forks.

Keywords

Cite

@article{arxiv.2403.13372,
  title  = {LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Models},
  author = {Yaowei Zheng and Richong Zhang and Junhao Zhang and Yanhan Ye and Zheyan Luo and Zhangchi Feng and Yongqiang Ma},
  journal= {arXiv preprint arXiv:2403.13372},
  year   = {2024}
}

Comments

13 pages, accepted to ACL 2024 System Demonstration Track