ai-agent

LlamaFactory: Fine-Tune AI Models

Name: LlamaFactory: Fine-Tune AI Models
Author: hiyouga

Fine-tune 100+ AI models with LlamaFactory, for founders. 73k+ GitHub stars.

73,131 stars8,936 forksPythonQuality 8/10Updated 7/8/2026100% free · open source

What it does

LlamaFactory allows founders to fine-tune over 100 pre-trained AI models for their specific use cases, enabling more accurate and efficient AI-powered solutions.

Install / run

git clone https://github.com/hiyouga/LlamaFactory.git && cd LlamaFactory

When to use it

•When you need to adapt a pre-trained AI model to your specific industry or task
•When you want to improve the performance of an AI model on a custom dataset
•When you need to deploy AI models in a production environment with minimal expertise

Quick start

1Run `pip install -r requirements.txt` to install the necessary dependencies
2Modify the `config.json` file to specify the model and dataset you want to use
3Use the `[train.py](https://github.com/hiyouga/LlamaFactory/blob/main/train.py)` script to fine-tune the model, e.g. `python train.py --model_name t5-base --dataset your_dataset`
4Evaluate the fine-tuned model using the `evaluate.py` script, e.g. `python evaluate.py --model_name t5-base --dataset your_dataset`
5Deploy the fine-tuned model using the `deploy.py` script, e.g. `python deploy.py --model_name t5-base`

Ready-to-paste prompt

Fine-tune a T5-base model on your custom dataset with `python train.py --model_name t5-base --dataset your_dataset --batch_size 16 --epochs 5`

Heads up: Make sure you have the correct version of Python (>=3.8) and the necessary GPU resources to run the model fine-tuning process, as specified in the README

Saves to your device