diffusers/tiny-stable-diffusion-torch

diffusers

Texto a imagen

Modelo pequeño de Stable Diffusion para generación de imágenes a partir de texto, publicado en Hugging Face y duplicado desde `hf-internal-testing/tiny-stable-diffusion-torch`. Está pensado principalmente para pruebas, integración y validación de flujos con Diffusers, no como modelo de producción de alta calidad visual.

Como usar

Instalación y uso con Diffusers:
pip install -U diffusers transformers accelerate

import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained(
    "diffusers/tiny-stable-diffusion-torch",
    dtype=torch.bfloat16,
    device_map="cuda"
)

prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]

Uso alternativo con StableDiffusionPipeline:
from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained(
    "hf-internal-testing/tiny-stable-diffusion-torch"
)

Funcionalidades

Tarea: generación de imagen desde texto.
Compatible con Diffusers y `StableDiffusionPipeline`.
Licencia Apache 2.0.
Modelo ligero tipo Tiny Stable Diffusion útil para pruebas rápidas de carga, inferencia y pipelines.
No aparece desplegado en proveedores de inferencia de Hugging Face en la página proporcionada.

Casos de uso

Probar que una integración con Hugging Face Diffusers carga modelos correctamente.
Validar código de inferencia de texto a imagen sin descargar un modelo grande.
Ejecutar pruebas automatizadas de pipelines Stable Diffusion.
Experimentar localmente con una versión mínima de Stable Diffusion.