ai-toolkit est un dépôt GitHub à but recherche, créé par Ostris, principalement utilisé pour l'expérimentation et l'entraînement du modèle Stable Diffusion. Il contient divers scripts IA, prenant en charge l'entraînement de modèles, la génération d'images et l'extraction de LoRA, entre autres. Cet outil est encore en développement et peut présenter une certaine instabilité, mais il offre des fonctionnalités riches et une grande personnalisation.