All 29 Dependencies
Every package ai-toolkit depends on, ranked by repo health score.
AI Toolkit by Ostris is a comprehensive, community-driven training framework for finetuning modern diffusion models without requiring expensive cloud infrastructure. It supports an extensive roster of image, video, and audio models—from FLUX.1/2 and SDXL to Wan 2.x video models and ACE-Step audio—all from a single, unified codebase.
The toolkit ships both a command-line interface driven by YAML config files and a full web UI built on Next.js that lets you start, stop, and monitor training jobs in a browser. This dual-mode design means researchers can script headless training pipelines on remote servers while practitioners can point-and-click their way through LoRA training on a local workstation.
At its core, AI Toolkit wraps Hugging Face Diffusers and Accelerate with a plugin-style extension system. Built-in extensions cover the most common workflows out of the box—LoRA, DoRA, LyCORIS, concept sliders, dataset captioning, and full fine-tuning—while the extension API lets advanced users add custom training pipelines without forking the core.
The project is actively maintained with multiple commits per day and a thriving Discord community. Cloud GPU deployment is supported on RunPod, Modal, and the author's own Ostris Cloud, making it straightforward to scale from a local 24 GB GPU to multi-node training when needed.
