trpfrog-diffusion

Primero instale los paquetes necesarios.

pip install diffusers transformers accelerate scipy safetensors

Luego, la imagen se puede generar con el siguiente script.

import torch
from diffusers import StableDiffusionPipeline

keyword = "trpfrog"
prompt = f"a photo of {keyword}"

model_id = "Prgckwb/trpfrog-diffusion"

pipe = StableDiffusionPipeline.from_pretrained(
model_id, torch_dtype=torch.float16
).to("cuda")

image = pipe(prompt).images[0]
image.save("trpfrog.jpg")