Repolode
kohya-ss

kohya-ss/sd-scripts

Scripts de treinamento, geração e utilitários para Stable Diffusion e outros modelos de geração de imagens, com suporte a LoRA e fine-tuning.

Estrelas
7,243
Forks
1,224
Issues abertas
917
Último commit
há 19 horas

O que é?

sd-scripts é um conjunto de scripts Python para treinar e executar modelos de geração de imagens da família Stable Diffusion. Oferece suporte ao treinamento de LoRA, fine-tuning completo (treinamento nativo e DreamBooth), Textual Inversion para SD e SDXL e treinamento de modelos de inpainting para SD1.5 e SDXL. As arquiteturas compatíveis incluem SD 1.x/2.x, SDXL, SD3/SD3.5, FLUX.1, Lumina, HunyuanImage-2.1, Anima e MOSS-Image.

Além do treinamento, o repositório inclui scripts de geração de imagens, ferramentas de cache de datasets, marcação de imagens com base no WD14, conversão de modelos e utilitários para mesclar LoRAs. O treinamento é configurado por meio de argumentos de linha de comando e arquivos de configuração de dataset, e pode ser executado no Windows e no Linux/WSL2 com PyTorch 2.6 ou posterior, usando Hugging Face Accelerate para iniciar os processos.

O projeto oferece documentação extensa em inglês e japonês que aborda o fluxo de treinamento de cada modelo, a configuração de datasets e opções avançadas, como deslocamentos de amostragem de timestep, masked loss e treinamento de ControlNet-LLLite.

Recursos principais

  • 01Treina redes LoRA, incluindo os módulos LoHa, LoKr, OFTv2 e BOFT.
  • 02Oferece suporte a fine-tuning completo e ao treinamento no estilo DreamBooth para a maioria dos modelos listados.
  • 03Oferece treinamento de Textual Inversion para SD e SDXL e treinamento de inpainting para SD1.5 e SDXL.
  • 04Inclui scripts de geração de imagens e utilitários para conversão de modelos e mesclagem de LoRAs.
  • 05Oferece arquivos de configuração de dataset para legendas, subconjuntos e candidatos a múltiplas legendas.

Casos de uso

  • Treinar uma LoRA personalizada de estilo ou personagem para SDXL ou FLUX.1 usando um dataset pessoal de imagens.
  • Fazer fine-tuning de um checkpoint de Stable Diffusion com um conjunto selecionado de imagens.
  • Gerar imagens em lote a partir de arquivos de prompts usando scripts de linha de comando.
  • Marcar um conjunto de imagens de treinamento com WD14 antes do treinamento.

Início rápido

  1. 1

    Siga as etapas de instalação para sua plataforma (Windows ou Linux/WSL2) no README.

Antes de usar

  • A instalação varia conforme a plataforma, e a configuração do DeepSpeed é experimental e está disponível apenas no Linux ou no WSL2.
  • Alguns recursos precisam de pacotes adicionais que não estão em requirements.txt, como scipy para os lms samplers e triton para os MOSS-Image int8 kernels.

Tópicos

Perguntas frequentes

O que é sd-scripts?
Scripts de treinamento, geração e utilitários para Stable Diffusion e outros modelos de geração de imagens, com suporte a LoRA e fine-tuning. sd-scripts é um conjunto de scripts Python para treinar e executar modelos de geração de imagens da família Stable Diffusion. Oferece suporte ao treinamento de LoRA, fine-tuning completo (treinamento nativo e DreamBooth), Textual Inversion para SD e SDXL e treinamento de modelos de inpainting para SD1.5 e SDXL. As arquiteturas compatíveis incluem SD 1.x/2.x, SDXL, SD3/SD3.5, FLUX.1, Lumina, HunyuanImage-2.1, Anima e MOSS-Image.
sd-scripts é gratuito? Qual é a licença?
sd-scripts é open source sob a licença Apache-2.0. Confira os termos da licença antes de usá-lo em produtos comerciais.
sd-scripts ainda é mantido?
Manutenção ativa. O último commit de sd-scripts foi em 18 de out. de 2026.
Para que posso usar sd-scripts?
Treinar uma LoRA personalizada de estilo ou personagem para SDXL ou FLUX.1 usando um dataset pessoal de imagens. Fazer fine-tuning de um checkpoint de Stable Diffusion com um conjunto selecionado de imagens. Gerar imagens em lote a partir de arquivos de prompts usando scripts de linha de comando. Marcar um conjunto de imagens de treinamento com WD14 antes do treinamento.
Como instalar sd-scripts?
Siga as etapas de instalação para sua plataforma (Windows ou Linux/WSL2) no README.
Onde consigo sd-scripts?
O código-fonte e as instruções de instalação estão no GitHub: https://github.com/kohya-ss/sd-scripts

Resumo gerado por IA a partir do README do projeto. As estatísticas do GitHub são atualizadas diariamente.

Projetos relacionados