sd-pokemon-diffusers

lambdalabs

Texto a imagen

Stable Diffusion ajustado en Pokémon por Lambda Labs. Introduce un texto como entrada y genera tu propio personaje Pokémon, ¡no es necesario 'ingeniería de prompts'! Si quieres averiguar cómo entrenar tus propias variantes de Stable Diffusion, consulta este ejemplo de Lambda Labs. Entrenado en imágenes de Pokémon con subtítulos BLIP usando 2xA6000 GPUs en Lambda GPU Cloud durante alrededor de 15,000 pasos (aproximadamente 6 horas, a un costo de alrededor de $10). Entrenado por Justin Pinkney (@Buntworthy) en Lambda Labs.

Como usar

!pip install diffusers==0.3.0
!pip install transformers scipy ftfy

import torch
from diffusers import StableDiffusionPipeline
from torch import autocast

pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/sd-pokemon-diffusers", torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "Yoda"
scale = 10
n_samples = 4

disable_safety = False

if disable_safety:
    def null_safety(images, **kwargs):
        return images, False
    pipe.safety_checker = null_safety

with autocast("cuda"):
    images = pipe(n_samples*[prompt], guidance_scale=scale).images

for idx, im in enumerate(images):
    im.save(f"{idx:06}.png")

Funcionalidades

Ajuste de Stable Diffusion en Pokémon
Generación de personajes Pokémon a partir de descripciones textuales
No se requiere ingeniería de prompts
Uso de GPUs 2xA6000 en Lambda GPU Cloud
Entrenamiento durante aproximadamente 15,000 pasos
Costo de entrenamiento alrededor de $10

Casos de uso

Generación de personajes personalizados de Pokémon
Creación de arte basado en descripciones textuales
Proyectos creativos y artísticos
Investigación en modelos de difusión y generación de imágenes