Repolode
Предложить
ostris

ostris/ai-toolkit

Комплексный набор инструментов для дообучения диффузионных моделей изображений, видео и аудио на потребительских GPU через веб-интерфейс или CLI.

Открыть на GitHub
Звёзды
12,378
Форки
1,587
Открытые issues
92
Последний коммит
позавчера

Что это?

AI Toolkit — набор инструментов для дообучения диффузионных моделей. Он поддерживает модели для генерации изображений, редактирования по инструкциям, видео и аудио, включая семейства FLUX, Qwen-Image, SDXL, Wan 2.x и LTX-2. Проект рассчитан на запуск этих моделей на потребительском оборудовании.

Обучением можно управлять через веб-интерфейс: запускать и останавливать задачи, а также следить за их ходом. Другой вариант — командная строка, где для запуска run.py передаются YAML-файлы конфигурации. После прерывания обучение возобновляется с последнего чекпойнта. Датасеты представляют собой папки с изображениями и соответствующими им файлами подписей .txt. Изображения автоматически масштабируются и распределяются по группам, поэтому обрезать их вручную не нужно.

Проект поддерживает сети LoRA и LoKr, а также выбор конкретных слоёв с помощью фильтров включения и исключения. Установить его можно через встроенный менеджер, который настраивает окружение PyTorch, или вручную — с Python 3.10+ и GPU Nvidia.

Ключевые возможности

  • 01Обучает модели диффузии для генерации и редактирования изображений, видео и аудио в одном наборе инструментов.
  • 02Предоставляет веб-интерфейс для запуска и остановки задач обучения и наблюдения за их ходом.
  • 03Поддерживает обучение через командную строку с YAML-файлами конфигурации.
  • 04Поддерживает сети LoRA и LoKr с фильтрами включения и исключения отдельных слоёв.
  • 05После прерывания возобновляет обучение с последнего чекпойнта.
  • 06Автоматически масштабирует изображения с разным соотношением сторон и распределяет их по группам.

Сценарии использования

  • Обучение LoRA для собственного стиля или персонажа на базе FLUX или SDXL с одной потребительской GPU.
  • Дообучение видеомодели Wan или LTX на собственном датасете.
  • Запуск задач обучения на удалённом сервере и наблюдение за ними через веб-интерфейс.

Быстрый старт

  1. 1

    Клонируйте репозиторий

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

    Перейдите в папку проекта

    cd ai-toolkit
  3. 3

    Запустите AI Toolkit Manager в Linux (он настроит окружение и откроет интерфейс по адресу http://localhost:8675)

    ./run_linux.sh

Что важно знать

  • Автор пометил менеджер с установкой в один клик как экспериментальный.
  • Для ручной установки требуется GPU Nvidia и достаточно VRAM для обучения выбранной модели.
  • Для некоторых поддерживаемых моделей, например FLUX.1-dev, сначала нужно запросить доступ на Hugging Face.

Темы

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

Частые вопросы

Что такое ai-toolkit?
Комплексный набор инструментов для дообучения диффузионных моделей изображений, видео и аудио на потребительских GPU через веб-интерфейс или CLI. AI Toolkit — набор инструментов для дообучения диффузионных моделей. Он поддерживает модели для генерации изображений, редактирования по инструкциям, видео и аудио, включая семейства FLUX, Qwen-Image, SDXL, Wan 2.x и LTX-2. Проект рассчитан на запуск этих моделей на потребительском оборудовании.
ai-toolkit можно использовать бесплатно? Какая у него лицензия?
ai-toolkit распространяется с открытым исходным кодом под лицензией MIT. Перед использованием в коммерческих продуктах проверьте условия лицензии.
ai-toolkit всё ещё поддерживается?
Активно поддерживается. Последний коммит в ai-toolkit — 19 окт. 2026 г..
Для чего можно использовать ai-toolkit?
Обучение LoRA для собственного стиля или персонажа на базе FLUX или SDXL с одной потребительской GPU. Дообучение видеомодели Wan или LTX на собственном датасете. Запуск задач обучения на удалённом сервере и наблюдение за ними через веб-интерфейс.
Как установить ai-toolkit?
Клонируйте репозиторий: git clone https://github.com/ostris/ai-toolkit.git Перейдите в папку проекта: cd ai-toolkit Запустите AI Toolkit Manager в Linux (он настроит окружение и откроет интерфейс по адресу http://localhost:8675): ./run_linux.sh
Где скачать ai-toolkit?
Исходный код и инструкции по установке — на GitHub: https://github.com/ostris/ai-toolkit

Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.

Похожие проекты