AI Toolkit is MIT licensed · Ostris Cloud is the hosted version

Train diffusion models.
Locally or in the cloud.

AI Toolkit is the all-in-one suite for finetuning image, video and audio diffusion models. Ostris Cloud runs it on rented GPUs with nothing to install.

Install AI Toolkit Launch on Ostris Cloud

Toolkit

github.com/ostris/ai-toolkit

AI ToolkitOpen source training for diffusion models. LoRA and LoKr, web UI and CLI, on your own GPU.

AI ToolkitOpen source · MIT

FLUX, SDXL, Qwen-Image, HiDream, Wan, LTX-2, Ace Step and more. Install it and train on your own hardware.Get started →

Ostris CloudHosted · pay per hour

The same AI Toolkit on rented GPUs. Upload a dataset, pick a model, train. Nothing to install.cloud.ostris.com →

Research notes

All posts →

Skipping SD3.5-large Blocks2024-10-23 FLUX IP Adapter Style Weights2024-09-12 Skipping FLUX.1-dev Blocks2024-09-07

Train diffusion models.Locally or in the cloud.

Research notes

Train diffusion models.
Locally or in the cloud.