训练与微调MITPython

AI Toolkit

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

Star 数
12.3K
Fork 数
1.6K
未关闭 Issue
97
最近提交
2026-10-14

数据来自 GitHub API · 更新于 2026-10-15

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.

主要功能

  • FLUX and video model LoRA training
  • Web UI for jobs and datasets
  • Config-driven runs

系统要求

平台
Linux · Windows
显卡
NVIDIA
显存
24 GB recommended for FLUX
Python
3.10+

如何安装 AI Toolkit

根据项目 README 整理。开源项目更新很快,请以 GitHub 页面上的最新说明为准。

From source

  1. 1
    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
  2. 2
    Run a training config.
    python run.py config/your_config.yml

优势

  • Go-to FLUX LoRA trainer
  • MIT license

不足

  • High VRAM for FLUX

AI Toolkit 替代品对比

软件Star 数许可证平台显存
AI Toolkit

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

12.3KMIT24 GB recommended for FLUX
Kohya’s GUI

GUI for training LoRA, DreamBooth and fine-tunes for SD, SDXL and FLUX.

12.6KApache-2.08 GB+ for SD1.5 LoRA; 12–24 GB for SDXL/FLUX
OneTrainer

One-stop trainer for diffusion models with a desktop UI.

3.2KAGPL-3.08 GB+
Unsloth

Fine-tune LLMs 2× faster with up to 70% less memory.

78KApache-2.03 GB+ depending on model size
Axolotl

Config-driven LLM fine-tuning for many models and techniques.

12.6KApache-2.0—
LLaMA-Factory

Unified fine-tuning for 100+ LLMs and VLMs with a no-code web UI.

75.5KApache-2.0—

相关模型

常见问题

AI Toolkit 免费吗?

是的——AI Toolkit 是开源软件,采用 MIT 许可证。商用前请确认软件及模型权重的许可条款。

AI Toolkit 支持 Windows、Mac 和 Linux 吗?

AI Toolkit 支持:Linux, Windows。

运行 AI Toolkit 需要什么配置?

NVIDIA. VRAM: 24 GB recommended for FLUX

如何安装 AI Toolkit?

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

AI Toolkit 有哪些好的替代品?

常见替代品包括 Kohya’s GUI, OneTrainer, Unsloth, Axolotl,详见本页对比表。