katuni4ka/tiny-random-stable-diffusion-with-safety-checker
katuni4ka
Texto a imagen
Modelo pequeño y aleatorio de Stable Diffusion para generación de texto a imagen, empaquetado como `StableDiffusionPipeline` con comprobador de seguridad. Está publicado en Hugging Face con Diffusers, Safetensors y licencia Apache 2.0; parece orientado a pruebas, integración y validación rápida más que a generación visual de alta calidad.
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(
"katuni4ka/tiny-random-stable-diffusion-with-safety-checker",
dtype=torch.bfloat16,
device_map="cuda"
)
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]
La página también muestra un ejemplo relacionado con StableDiffusionPipeline:
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained("hf-internal-testing/tiny-stable-diffusion-torch")
Funcionalidades
- Generación de imágenes a partir de prompts de texto.
- Compatible con la biblioteca Diffusers de Hugging Face.
- Usa el pipeline `StableDiffusionPipeline`.
- Incluye comprobador de seguridad.
- Distribuido en formato Safetensors.
- No está desplegado en proveedores de inferencia de Hugging Face en la página indicada.
- Tuvo 1.752 descargas el último mes según la página proporcionada.
Casos de uso
- Pruebas rápidas de pipelines de Stable Diffusion con Diffusers.
- Validación de integración para aplicaciones que cargan modelos desde Hugging Face.
- Comprobación de flujos de texto a imagen que necesitan un modelo pequeño.
- Ensayos de compatibilidad con Safetensors, CUDA o dispositivos Apple mediante MPS.
- Pruebas automatizadas donde importa más la estructura del pipeline que la calidad final de la imagen.