SDXS-512-DreamShaper

IDKiro

Texto a imagen

SDXS es un modelo que puede generar imágenes de alta resolución en tiempo real basado en textos de indicaciones, entrenado utilizando destilación de puntajes y correspondencia de características. Para más información, por favor refiérase a nuestro artículo de investigación: SDXS: Modelos de Difusión Latente en un Paso en Tiempo Real con Condiciones de Imagen. Abrimos el modelo como parte de la investigación. SDXS-512-DreamShaper es la versión que entrenamos específicamente para la comunidad.

Como usar

import torch
from diffusers import StableDiffusionPipeline, AutoencoderKL

repo = "IDKiro/sdxs-512-dreamshaper"
seed = 42
weight_type = torch.float16 # o float32

# Cargar el modelo.
pipe = StableDiffusionPipeline.from_pretrained(repo, torch_dtype=weight_type)
pipe.to("cuda")

prompt = "una imagen primer plano de un anciano parado bajo la lluvia"

# Asegúrese de usar 1 paso de inferencia y CFG establecido en 0.
image = pipe(
prompt, 
num_inference_steps=1, 
guidance_scale=0,
generator=torch.Generator(device="cuda").manual_seed(seed)
).images[0]

image.save("output.png")

Funcionalidades

Generación de imágenes de alta resolución en tiempo real
Entrenado utilizando destilación de puntajes y correspondencia de características
Diseñado específicamente para la comunidad
Enfocado en la calidad de generación de imágenes sacrificando la diversidad

Casos de uso

Generación de imágenes a partir de textos descriptivos
Aplicaciones artísticas donde se necesita calidad en cada generación de imagen
Proyectos comunitarios que requieren imágenes en alta resolución