This model is a diffusion model for unconditional image generation of mammograms of size 64x64 pixels.
The model was trained with 1000 images using the
DDPM architecture.
The model was trained for 50 epochs with a batch size of 64, using around 11 GB of GPU memory.
1from diffusers import DDPMPipeline
2
3pipeline = DDPMPipeline.from_pretrained({hub_model_id})
4image = pipeline().images[0]
5image
