Executando verificação de segurança...
1

[Dúvida] Existe alguma forma de usar um notebook do google colab como um "mini backend" para um projeto de Stable Diffusion?

Olá, atualmente estou executando a implementação do Stable Diffusion no Google Colab para geração de imagens simples.

Aqui está o código que utilizei no Google Colab:

!pip install diffusers==0.11.1
!pip install transformers scipy ftfy accelerate

import torch
from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)

pipe = pipe.to("cuda")

prompt = "make a minimalistic and abstract logo for a website about 3D solids, very colorful and harmonic"
image = pipe(prompt).images[0]
image

É importante ressaltar que esta é uma implementação pessoal e, por enquanto, não estou planejando criar um backend completo para hospedar esse serviço. Estou explorando essa abordagem como um projeto pessoal e, dependendo do resultado final, talvez possa utilizá-la em um trabalho acadêmico futuro. Eu estava em busca de uma maneira de realizar uma requisição ao notebook do Colab através de uma URL e obter a imagem resultante como retorno. Infelizmente, minhas pesquisas não revelaram uma solução simples e de código aberto para esse propósito. Alguma ideia?

Carregando publicação patrocinada...
1