Repolode
提交项目
ostris

ostris/ai-toolkit

面向扩散模型微调的一体化训练套件，支持图像、视频和音频模型，可通过 Web UI 或命令行使用。

在 GitHub 查看
Star
12,378
Fork
1,587
未关闭 Issue
92
最近提交
21小时前

这是什么？

AI Toolkit 是一套用于微调扩散模型的训练套件，覆盖图像、基于指令的图像编辑、视频和音频模型，包括 FLUX、Qwen-Image、SDXL、Wan 2.x、LTX-2 等系列。项目目标是让这些模型能在消费级显卡上训练。

训练既可以通过 Web UI 完成，在界面中启动、停止和监控任务；也可以用命令行，通过 YAML 配置文件运行 run.py。训练中断后会从最近的检查点继续。数据集是一个包含图片及同名 .txt 描述文件的文件夹，图片会自动缩放并分桶，无需手动裁剪。

项目支持 LoRA 和 LoKr 网络，并可通过过滤规则指定训练或忽略的层。安装方式包括内置的管理器（自动配置 PyTorch 环境），也可以手动安装，需要 Python 3.10 及以上和 Nvidia 显卡。

核心功能

  • 01在同一套工具中训练图像、图像编辑、视频和音频扩散模型。
  • 02提供 Web UI，用于启动、停止和监控训练任务。
  • 03支持通过 YAML 配置文件进行命令行训练。
  • 04支持 LoRA 和 LoKr 网络，可按层设置包含或排除规则。
  • 05训练中断后可从最近的检查点继续。
  • 06自动缩放图片并按宽高比分桶，无需手动裁剪。

适用场景

  • 在一块消费级显卡上为 FLUX 或 SDXL 训练自定义风格或角色的 LoRA。
  • 用个人数据集微调 Wan 或 LTX 视频模型。
  • 在远程服务器上运行训练任务，并通过 Web UI 监控进度。

快速上手

  1. 1

    克隆仓库

    git clone https://github.com/ostris/ai-toolkit.git
  2. 2

    进入项目文件夹

    cd ai-toolkit
  3. 3

    在 Linux 上启动 AI Toolkit Manager（自动配置环境并在 http://localhost:8675 启动界面）

    ./run_linux.sh

使用前须知

  • 作者标注一键管理器仍为实验性功能。
  • 手动安装需要 Nvidia GPU，并且显存要足以支持所训练的模型。
  • 部分支持的模型（如 FLUX.1-dev）需要先在 Hugging Face 申请访问权限。

标签

#diffusion-models#lora#fine-tuning#stable-diffusion#flux#video-generation#pytorch

常见问题

ai-toolkit 是什么？
面向扩散模型微调的一体化训练套件，支持图像、视频和音频模型，可通过 Web UI 或命令行使用。 AI Toolkit 是一套用于微调扩散模型的训练套件，覆盖图像、基于指令的图像编辑、视频和音频模型，包括 FLUX、Qwen-Image、SDXL、Wan 2.x、LTX-2 等系列。项目目标是让这些模型能在消费级显卡上训练。
ai-toolkit 可以免费使用吗？用的什么许可证？
ai-toolkit 以 MIT 许可证开源。用于商业产品前，请先确认许可证条款。
ai-toolkit 还在维护吗？
活跃维护中。ai-toolkit 最近一次提交是在 2026年10月19日。
ai-toolkit 适合用来做什么？
在一块消费级显卡上为 FLUX 或 SDXL 训练自定义风格或角色的 LoRA。 用个人数据集微调 Wan 或 LTX 视频模型。 在远程服务器上运行训练任务，并通过 Web UI 监控进度。
ai-toolkit 怎么安装和运行？
克隆仓库: git clone https://github.com/ostris/ai-toolkit.git 进入项目文件夹: cd ai-toolkit 在 Linux 上启动 AI Toolkit Manager（自动配置环境并在 http://localhost:8675 启动界面）: ./run_linux.sh
在哪里获取 ai-toolkit？
源代码和安装说明在 GitHub 上：https://github.com/ostris/ai-toolkit

概要由 AI 根据项目 README 生成；GitHub 数据每日更新。

相关项目