Scripts de treinamento, geração e utilitários para Stable Diffusion e outros modelos de geração de imagens, incluindo LoRA e fine-tuning.
7,2 mil1,2 milPythonApache-2.0Atualizado há 3 dias
ostris/ai-toolkit
Suite completa para ajustar modelos de difusão de imagem, vídeo e áudio em GPUs de consumo, pela interface web ou CLI.
- Estrelas
- 12,378
- Forks
- 1,587
- Issues abertas
- 92
- Último commit
- anteontem
O que é?
AI Toolkit é uma suite para ajustar modelos de difusão. Ela abrange modelos de imagem, edição baseada em instruções, vídeo e áudio, incluindo famílias como FLUX, Qwen-Image, SDXL, Wan 2.x e LTX-2. O projeto busca permitir que esses modelos sejam executados em hardware de consumo.
É possível controlar o treinamento por uma interface web, que inicia, interrompe e monitora jobs, ou pela linha de comando, usando arquivos de configuração YAML passados para run.py. Após uma interrupção, o treinamento continua a partir do último checkpoint. Os datasets são pastas de imagens com arquivos .txt correspondentes contendo as legendas; as imagens são redimensionadas e agrupadas automaticamente, sem necessidade de corte manual.
O projeto oferece suporte a redes LoRA e LoKr, com opções para selecionar camadas específicas usando filtros de inclusão e exclusão. A instalação pode ser feita por um gerenciador integrado que configura o ambiente PyTorch ou manualmente, com Python 3.10+ e uma GPU Nvidia.
Recursos principais
- 01Treina modelos de difusão de imagem, edição, vídeo e áudio em um único toolkit.
- 02Oferece uma interface web para iniciar, interromper e monitorar jobs de treinamento.
- 03Permite treinar pela linha de comando usando arquivos de configuração YAML.
- 04Oferece suporte a redes LoRA e LoKr com filtros de inclusão e exclusão por camada.
- 05Retoma o treinamento do último checkpoint após uma interrupção.
- 06Redimensiona e agrupa automaticamente imagens com diferentes proporções.
Casos de uso
- Treinar uma LoRA personalizada de estilo ou personagem para FLUX ou SDXL em uma única GPU de consumo.
- Ajustar um modelo de vídeo Wan ou LTX usando um dataset próprio.
- Executar jobs de treinamento em um servidor remoto e monitorá-los pela interface web.
Início rápido
- 1
Clone o repositório
git clone https://github.com/ostris/ai-toolkit.git - 2
Acesse a pasta do projeto
cd ai-toolkit - 3
Inicie o AI Toolkit Manager no Linux (ele configura o ambiente e abre a interface em http://localhost:8675)
./run_linux.sh
Antes de usar
- O autor classifica o gerenciador de um clique como experimental.
- A instalação manual requer uma GPU Nvidia e VRAM suficiente para o modelo que você vai treinar.
- Alguns modelos compatíveis, como FLUX.1-dev, exigem que você solicite acesso primeiro no Hugging Face.
Tópicos
Perguntas frequentes
- O que é ai-toolkit?
- Suite completa para ajustar modelos de difusão de imagem, vídeo e áudio em GPUs de consumo, pela interface web ou CLI. AI Toolkit é uma suite para ajustar modelos de difusão. Ela abrange modelos de imagem, edição baseada em instruções, vídeo e áudio, incluindo famílias como FLUX, Qwen-Image, SDXL, Wan 2.x e LTX-2. O projeto busca permitir que esses modelos sejam executados em hardware de consumo.
- ai-toolkit é gratuito? Qual é a licença?
- ai-toolkit é open source sob a licença MIT. Confira os termos da licença antes de usá-lo em produtos comerciais.
- ai-toolkit ainda é mantido?
- Manutenção ativa. O último commit de ai-toolkit foi em 19 de out. de 2026.
- Para que posso usar ai-toolkit?
- Treinar uma LoRA personalizada de estilo ou personagem para FLUX ou SDXL em uma única GPU de consumo. Ajustar um modelo de vídeo Wan ou LTX usando um dataset próprio. Executar jobs de treinamento em um servidor remoto e monitorá-los pela interface web.
- Como instalar ai-toolkit?
- Clone o repositório: git clone https://github.com/ostris/ai-toolkit.git Acesse a pasta do projeto: cd ai-toolkit Inicie o AI Toolkit Manager no Linux (ele configura o ambiente e abre a interface em http://localhost:8675): ./run_linux.sh
- Onde consigo ai-toolkit?
- O código-fonte e as instruções de instalação estão no GitHub: https://github.com/ostris/ai-toolkit
Resumo gerado por IA a partir do README do projeto. As estatísticas do GitHub são atualizadas diariamente.
Projetos relacionados
GUI em grafo de nós, API e backend para executar modelos de difusão localmente e gerar imagens, vídeos, áudio e conteúdo 3D.
IA e Machine Learning#stable-diffusion#diffusion-models#node-editor#image-generation#video-generation
140,6 mil16,6 milPythonGPL-3.0Atualizado há 20 horas
Um mecanismo open source de modelos Diffusion para gerar imagens, vídeos e áudio, com suporte a inferência e treinamento com pouca VRAM.
13,3 mil1,3 milPythonApache-2.0Atualizado ontem
Biblioteca Python que executa LLMs grandes camada por camada, permitindo inferência com modelos da classe 70B em uma única GPU de 4 GB, sem quantização.
39,5 mil4 milJupyter NotebookApache-2.0Atualizado anteontem
Biblioteca PyTorch de modelos de difusão pré-treinados, schedulers de ruído e componentes para geração de imagens, vídeos e áudio.
34,8 mil7,4 milPythonApache-2.0Atualizado há 17 horas
Framework gratuito e open source para fine-tuning e pós-treinamento de LLMs, configurado por arquivos YAML.
12,6 mil1,6 milPythonApache-2.0Atualizado ontem