metadata

library_name: diffusers
pipeline_tag: text-to-image
tags:
  - text-to-image
  - image-generation
license: mit
datasets:
  - your-dataset-name
base_model: stabilityai/stable-diffusion-2-1
thumbnail: https://your-thumbnail-link.com/thumbnail.png

Nombre del Modelo

Este modelo está diseñado para generar imágenes a partir de descripciones de texto utilizando la arquitectura Stable Diffusion.

Descripción

Este modelo fue entrenado utilizando el conjunto de datos your-dataset-name y está basado en el modelo stabilityai/stable-diffusion-2-1. Está optimizado para tareas de generación de imágenes a partir de texto, utilizando la librería diffusers.

Cómo Usar

Puedes usar este modelo directamente con la Inference API de Hugging Face. Aquí tienes un ejemplo en Python:

from diffusers import StableDiffusionPipeline
import torch

model_id = "tu-usuario/tu-modelo"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "Una pintura surrealista de un gato volando en el espacio"
image = pipe(prompt).images[0]

image.save("output.png")