Como executar difusão estável no Google Colab

Stable Diffusion é um poderoso modelo generativo que foi treinado em grandes conjuntos de dados de imagens e vídeos para criar amostras realistas e de alta qualidade. Ele funciona refinando iterativamente um sinal de entrada de ruído para gerar imagens com níveis crescentes de detalhes e coerência. Essa técnica tem sido amplamente utilizada no campo da arte gerada por IA, permitindo que artistas e entusiastas criem obras de arte únicas e fascinantes.

O Google Colab, por outro lado, é uma plataforma baseada em nuvem que fornece acesso gratuito a poderosas GPUs, tornando-se uma excelente opção para experimentar modelos de aprendizado de máquina. Ele também permite que os usuários colaborem e compartilhem seu trabalho com outras pessoas facilmente. Ao combinar o Stable Diffusion com o Google Colab, podemos acessar modelos pré-treinados e gerar nossa própria arte de IA com configuração e custo mínimos.

Neste tutorial, guiaremos você pelas etapas para começar a usar o Stable Diffusion no Google Colab, incluindo a configuração do ambiente, o carregamento de modelos pré-treinados e a geração de amostras de arte de IA. Seja você um iniciante ou um praticante experiente no campo da arte gerada por IA, este tutorial fornecerá o conhecimento essencial para começar a criar suas próprias peças de arte exclusivas.

Introdução à difusão estável

Para começar, faremos uma breve demonstração de todo o processo de execução do Stable Diffusion no Google Colab, até a geração de nosso conjunto inicial de imagens.

Conta no rosto de abraços

Para usar o Stable Diffusion, precisamos baixá-lo do Hugging Face, que é um repositório que fornece uma variedade de modelos e funcionalidades úteis. Para baixar o Stable Diffusion, primeiro precisamos criar uma conta no Hugging Face. Um modelo é essencialmente um programa que pode aprender a executar determinadas tarefas de forma independente.

Criar uma conta com o Hugging Face é um processo simples que envolve visitar o site do Hugging Face e seguir os procedimentos padrão de criação de conta. Depois de concluir o registro, você receberá um e-mail de confirmação para verificar sua conta. Depois de confirmar sua conta, você poderá baixar o Stable Diffusion e começar a experimentar a arte gerada por IA no Google Colab.

Copie o Notebook Stable Diffusion Colab para o Google Drive

Para começar, semelhante à edição de qualquer documento do Google criado por outra pessoa, você precisará visitar o notebook Stable Diffusion Google Colab no Google Collab e selecionar a opção "Arquivo" seguido de "Salvar uma cópia no Drive".

Isso abrirá uma nova guia com o notebook salvo em sua unidade e seu nome será "Cópia de difusão estável com difusores". Sinta-se à vontade para renomeá-lo conforme sua preferência.

Faça uso da GPU

Antes de executar o Stable Diffusion no Google Colab, é importante garantir que o notebook esteja configurado para usar uma GPU. A GPU é um componente essencial para o processamento eficiente de grandes conjuntos de dados e a execução de cálculos complexos envolvidos no treinamento de modelos generativos, como a difusão estável.

Para verificar se o notebook está configurado para usar uma GPU, acesse o menu "Runtime" e selecione "Alterar tipo de runtime". Uma pequena janela aparecerá com um menu suspenso "Acelerador de hardware". Certifique-se de que "GPU" esteja selecionado como o acelerador de hardware.

Depois de selecionar "GPU", clique em "Salvar" para aplicar as alterações. Agora, o notebook está configurado para usar uma GPU, o que nos permitirá executar o Stable Diffusion de maneira suave e eficiente. Com essa configuração, agora podemos carregar os modelos pré-treinados e gerar nossa própria arte de IA.

Execute as primeiras células

Agora podemos executar as células iniciais no notebook Stable Diffusion Colab. Para fazer isso, basta passar o mouse sobre cada célula e um botão de reprodução aparecerá. Clique no botão play e espere a célula terminar sua execução. Você saberá que uma célula concluiu sua tarefa quando uma marca de seleção verde aparecer.

Cada célula é acompanhada por uma descrição acima dela, explicando sua finalidade e funcionalidade. Se você encontrar uma mensagem como "este notebook requer muita RAM", basta clicar em "OK" para prosseguir.

Depois de executar com sucesso as células iniciais, você deve estar pronto para começar a usar a difusão estável. O notebook oferece uma interface amigável para carregar modelos pré-treinados e gerar arte gerada por IA. Quer você seja novo na área ou um praticante experiente, este caderno pode servir como uma ferramenta valiosa para explorar as possibilidades da difusão estável na geração de obras de arte únicas e de alta qualidade.

Execute a quinta célula para baixar os arquivos necessários

Continuaremos a executar a quinta célula, localizada na seção "Stable Diffusion Pipeline", que fará o download dos componentes necessários.

Além disso, precisamos executar a seguinte célula também.

pipe = pipe.to("cuda")

Gerar imagem

Na célula seguinte, você já deve ter notado uma imagem aparecendo. É aqui que vamos gerar nossa primeira imagem. Para isso, basta digitar uma string de texto entre aspas e executar a célula. A título de ilustração, incluí um exemplo onde inseri o seguinte texto:

a protoss cityscape with advanced technology, inspired by the game starcraft, making heavy use of light and shadow to create a sense of mystery and foreboding. the city sprawling below is a mix of organic and inorganic, with swirling energy currents and strange crystalline structures, illustrated in a realistic and detailed style by wei wang, artstationa protoss cityscape with advanced technology, inspired by the game starcraft, making heavy use of light and shadow to create a sense of mystery and foreboding. the city sprawling below is a mix of organic and inorganic, with swirling energy currents and strange crystalline structures, illustrated in a realistic and detailed style by wei wang, artstation

Bom trabalho! O processo de geração dessa imagem deveria ter levado menos de um minuto.

Empacotando

Parabéns por gerar sua primeira imagem usando Stable Diffusion de Hugging Face no Google Colab! Este é apenas o começo de sua jornada no excitante mundo da arte gerada por IA.

Ao seguir este guia, esperamos ter ajudado você a superar a barreira inicial e fornecido o conhecimento necessário para começar a explorar ainda mais a difusão estável. Incentivamos você a continuar explorando as outras instruções no bloco de anotações do Google Colab, o que permitirá que você personalize e ajuste seu modelo para suas necessidades específicas.

No entanto, entendemos que você pode encontrar alguns problemas ou ter dúvidas ao longo do caminho. Não hesite em deixar um comentário, e nossa equipe entrará em contato com você o mais breve possível. Estamos aqui para apoiá-lo e ajudá-lo a ter sucesso em sua jornada de arte gerada por IA. Continue experimentando e mal podemos esperar para ver o que você criará a seguir!