katuni4ka/tiny-stable-diffusion-torch-custom-variant
katuni4ka
Texto a imagen
Modelo de texto a imagen alojado en Hugging Face, compatible con Diffusers y la clase StableDiffusionPipeline. Es una variante personalizada pequeña de Stable Diffusion orientada a generar imágenes a partir de prompts de texto.
Como usar
Instalación y ejemplo de 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("katuni4ka/tiny-stable-diffusion-torch-custom-variant", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]
Funcionalidades
- Generación de imágenes a partir de texto mediante Diffusers.
- Compatible con DiffusionPipeline de la biblioteca diffusers.
- Uso local con PyTorch, transformers y accelerate.
- Soporte sugerido para ejecución en CUDA y cambio a MPS en dispositivos Apple.
- No está desplegado actualmente en proveedores de inferencia de Hugging Face.
Casos de uso
- Crear imágenes desde descripciones textuales en aplicaciones locales.
- Probar una variante pequeña/personalizada de Stable Diffusion con Diffusers.
- Experimentar con prompts visuales sin depender de un proveedor de inferencia alojado.