Ostris AI Toolkit

Обучение диффузионных моделей — LoRA, full fine-tune с визуальным интерфейсом.

Обучение моделей Linux от 12 ГБ VRAM

Для кого

ML-инженеры и художники, которым нужен гибкий инструмент для обучения LoRA и full fine-tune диффузионных моделей.

Что внутри

  • Ostris AI Toolkit — веб-интерфейс для тренировки моделей
  • Поддержка SDXL, FLUX и других диффузионных архитектур
  • Tensorboard для мониторинга
  • Jupyter для экспериментов

Как начать

  1. Запустите инстанс с этим шаблоном (рекомендуется 48+ ГБ диска)
  2. Откройте AI Toolkit на порту 8675
  3. Загрузите датасет и начните тренировку

Рекомендации по GPU

  • 12 ГБ VRAM — LoRA для SD 1.5
  • 24 ГБ VRAM (RTX 4090) — LoRA для SDXL и FLUX
  • 40+ ГБ VRAM — full fine-tune

Предустановленное ПО

Ubuntu 22.04 CUDA 12.9 AI Toolkit Tensorboard Python 3.11

Рекомендуемые GPU

RTX 4090 A100 40GB RTX 3090

Готовы запустить Ostris AI Toolkit?

Запустить GPU-сервер