How to use dimpo/ddpm-mnist with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("dimpo/ddpm-mnist", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
It creates images 32x32 manners... the mnist original size 28x28 pixels...
And. after investigating, I could see that some numbers are horizontally flipped... 7 looks like 5
· Sign up or log in to comment
