johnowhitaker's picture
Update README.md
5662182
|
raw
history blame
386 Bytes

Diffusion model trained on https://huggingface.co/datasets/huggan/smithsonian_butterflies_subset

Example outputs: examples

Usage:

pipeline = DDPMPipeline.from_pretrained('tglcourse/ddpm-ema-butterflies-64').to(device)
images = pipeline(batch_size=8).images

More info in course soon.