Herminia, modelo de generación de imágenes

Modelo herminia creado a partir de Stable Diffusion para generar imágenes a partir de texto.

Uso con GPU

from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("cz9dev/herminia", torch_dtype=torch.float16)
pipe.to("cuda")

image = pipe("Un astronauta montando un unicornio").images[0]
image.save("astronauta.png")

Uso con CPU

from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("cz9dev/herminia", torch_dtype=torch.float32)
pipe.to("cpu")

image = pipe("Arte fantástico con criaturas mitológicas").images[0]
image.save("astronauta.png")

Ejemplos de imágenes generadas

"Paisaje futurista de la tierra"

"Un dibujo abstracto de la bandera cubana"

Licencia

MIT