Training & fine-tuningMITPython

AI Toolkit

Ostris’ training suite for FLUX, video and image model LoRAs.

Live from the GitHub API · updated 2026-10-15

What is AI Toolkit?

AI Toolkit is a popular trainer for FLUX and newer image/video model LoRAs, driven by YAML configs and a web UI for managing jobs and datasets.

Summarized from the project’s README. Projects change quickly — check the GitHub page for the latest instructions.

Clone, create a venv and install (install PyTorch first per the README).

git clone https://github.com/ostris/ai-toolkit.git
cd ai-toolkit
python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt

Software	Stars	License	VRAM
AI Toolkit Ostris’ training suite for FLUX, video and image model LoRAs.	12.3K	MIT	24 GB recommended for FLUX
Kohya’s GUI GUI for training LoRA, DreamBooth and fine-tunes for SD, SDXL and FLUX.	12.6K	Apache-2.0	8 GB+ for SD1.5 LoRA; 12–24 GB for SDXL/FLUX
OneTrainer One-stop trainer for diffusion models with a desktop UI.	3.2K	AGPL-3.0	8 GB+
Unsloth Fine-tune LLMs 2× faster with up to 70% less memory.	78K	Apache-2.0	3 GB+ depending on model size
Axolotl Config-driven LLM fine-tuning for many models and techniques.	12.6K	Apache-2.0	—
LLaMA-Factory Unified fine-tuning for 100+ LLMs and VLMs with a no-code web UI.	75.5K	Apache-2.0	—

Is AI Toolkit free?

Yes — AI Toolkit is open-source software released under the MIT license. Check the license (and any model weight licenses) before commercial use.

Does AI Toolkit run on Windows, Mac and Linux?

AI Toolkit supports: Linux, Windows.

What hardware do I need for AI Toolkit?

NVIDIA. VRAM: 24 GB recommended for FLUX

How do I install AI Toolkit?

1. Clone, create a venv and install (install PyTorch first per the README). 2. Run a training config.

What are the best AI Toolkit alternatives?

Popular alternatives include Kohya’s GUI, OneTrainer, Unsloth, Axolotl. See the comparison table on this page.