waifu-diffusion

hakurei

Texto a imagen

waifu-diffusion es un modelo de difusión de texto a imagen latente que ha sido condicionado en imágenes de anime de alta calidad mediante ajuste fino. También soporta una interfaz web de Gradio y Colab con Diffusers para ejecutar Waifu Diffusion.

Como usar

import torch
from torch import autocast
from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained(
'hakurei/waifu-diffusion',
torch_dtype=torch.float32
).to('cuda')

prompt = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt"
with autocast("cuda"):
    image = pipe(prompt, guidance_scale=6)["sample"][0]  
    
image.save("test.png")

Funcionalidades

Modelo de difusión de texto a imagen latente
Condicionado con imágenes de anime de alta calidad
Interfaz web de Gradio
Soporte de Colab con Diffusers

Casos de uso

Usos para entretenimiento
Asistente de arte generativo