segmind/tiny-sd
segmind
Texto a imagen
Este pipeline fue destilado de SG161222/Realistic_Vision_V4.0 en un subconjunto del conjunto de datos recastai/LAION-art-EN-improved-captions. A continuación se presentan algunas imágenes de ejemplo generadas con el modelo tiny-sd. Hemos observado que los modelos destilados son hasta un 80% más rápidos que los modelos Base SD1.5.
Como usar
Puedes usar el pipeline de la siguiente manera:
from diffusers import DiffusionPipeline
import torch
pipeline = DiffusionPipeline.from_pretrained("segmind/tiny-sd", torch_dtype=torch.float16)
prompt = "Retrato de una chica bonita"
image = pipeline(prompt).images[0]
image.save("my_image.png")
Funcionalidades
- Modelo Text-to-Image
- Basado en la difusión estable
- Entrenado con captions mejoradas de recastai/LAION-art-EN
- Resolución de imagen de 512
- Precisión mixta: fp16
- Pasos: 125000
- Tasa de aprendizaje: 1e-4
- Tamaño de lote: 32
- Pasos de acumulación de gradiente: 4
Casos de uso
- Generación de imágenes realistas a partir de texto
- Creación de arte generativo