dimitribarbot/Z-Image-Turbo-BF16
dimitribarbot
Texto a imagen
Versión BF16 de Tongyi-MAI/Z-Image-Turbo para generación de imágenes a partir de texto. Reduce la descarga del transformador frente a la versión FP32, usando aproximadamente 12,3 GB en lugar de 24,6 GB, y evita tener que descargar el modelo FP32 para convertirlo después a BF16.
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("dimitribarbot/Z-Image-Turbo-BF16", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]
Funcionalidades
- Modelo de texto a imagen basado en Diffusers.
- Pesos en formato Safetensors.
- Pipeline compatible: ZImagePipeline / DiffusionPipeline.
- Versión BF16 optimizada para menor tamaño de descarga que FP32.
- Modelo derivado y cuantizado de Tongyi-MAI/Z-Image-Turbo.
- Licencia Apache 2.0, siguiendo además los términos y restricciones del modelo original.
Casos de uso
- Generación rápida de imágenes desde prompts de texto.
- Ejecución local de Z-Image-Turbo en BF16 para reducir tamaño de descarga y memoria frente a FP32.
- Experimentación con pipelines Diffusers para modelos de texto a imagen.
- Creación de imágenes estilizadas o detalladas a partir de descripciones textuales.