Treine seu próprio modelo DreamBooth com o Github Actions e Replicate
Tabela de conteúdos
Introdução
Olá, eu sou Zeke, um engenheiro da Replicate. Hoje, vou mostrar para você uma nova ferramenta que desenvolvemos para facilitar a publicação do seu próprio modelo do Dreambooth usando as ações do GitHub.
Recentemente, publicamos um post no blog sobre uma API que criamos, a qual torna muito simples treinar e publicar seu próprio modelo usando o Dreambooth. Neste post, apresentamos uma série de etapas que você deve seguir, utilizando o terminal de comando para criar um arquivo zip com seus dados de treinamento, fazer o upload desse arquivo para uma URL específica e, em seguida, utilizar nossa API para iniciar o treinamento e gerar um modelo.
Agora, com o GitHub Actions, você pode executar todo esse processo diretamente no GitHub, sem precisar escrever código. Vamos dar uma olhada neste repositório...
Passo 1: Preparação
Antes de começar, você precisa fazer o fork deste repositório em sua própria conta do GitHub. Basta clicar no botão "Fork" no canto superior direito da página.
Passo 2: Configuração do Repositório
Normalmente, você removeria as imagens fofas de cachorrinhos e as substituiria pelas suas próprias imagens personalizadas, mas, para fins deste Tutorial, vamos pular essa etapa.
Passo 3: Configurar as Chaves de Acesso
Agora, precisamos configurar uma chave de acesso para a API da Replicate.
Vá até a guia "Settings" (Configurações) do seu repositório e clique em "Secrets" (Segredos). Em seguida, clique em "New repository secret" (Novo segredo do repositório) e defina o nome do segredo como "replicate API token" (token da API da Replicate).
Nota: Certifique-se de não compartilhar esse token com mais ninguém.
Passo 4: Executar o Fluxo de Trabalho
Agora estamos prontos para executar o fluxo de trabalho. Vá até a guia "Actions" (Ações) do seu repositório e clique em "Train a Model" (Treinar um modelo). Em seguida, clique no botão verde "Run workflow" (Executar fluxo de trabalho).
Nesta tela, você verá campos para preencher com informações sobre o seu modelo. O primeiro campo é o nome do modelo, que deve começar com seu nome de usuário do GitHub, seguido por um nome de sua escolha. Por exemplo, se meu nome de usuário for Zeke, eu poderia nomear meu modelo como "dog booth" (cabine de cachorros). O próximo campo é um identificador exclusivo para o modelo. Este deve ser uma sequência de caracteres aleatória e não uma palavra real. Recomendamos que você escolha Algo semelhante a "dgg". Por fim, preencha o número de etapas de treinamento (o valor padrão é 2000, mas você pode ajustar isso se desejar).
Clique em "Run workflow" (Executar fluxo de trabalho) para iniciar o treinamento do modelo.
Passo 5: Acompanhar o Treinamento do Modelo
Aguarde alguns minutos enquanto o fluxo de trabalho executa. Você pode acompanhar o progresso do treinamento na página "Actions" (Ações). Quando o treinamento for concluído, você receberá um link para o seu novo modelo.
Passo 6: Utilizar o Modelo Treinado
Agora que o treinamento do modelo foi concluído, você pode utilizá-lo para gerar predições. Para isso, acesse a página do seu modelo e digite um Prompt para a geração de imagens. Por exemplo, você pode digitar "Um desenho de um dgg tocando piano". O modelo irá gerar uma imagem com base nesse prompt.
Passo 7: Compartilhar o Modelo
Por padrão, os modelos são privados, o que significa que apenas você pode utilizá-los. No entanto, você pode optar por tornar seu modelo público e permitir que outras pessoas o utilizem.
Basta acessar a página de configurações do seu modelo e alterar a opção de visibilidade para "Public" (Público).
Passo 8: Adicionar Exemplos à Galeria
Após gerar uma predição com o modelo e gostar do resultado, você pode adicioná-la à galeria de exemplos. Para isso, basta clicar em "Add to Example Gallery" (Adicionar à Galeria de Exemplos) na página de predição.
Conclusão
Agora você sabe como utilizar o GitHub Actions para treinar e publicar seu próprio modelo do Dreambooth. Divirta-se treinando seu modelo e, se tiver alguma dúvida, não hesite em nos contatar no Discord (discord.gg/replicate), onde temos uma comunidade pronta para ajudá-lo.