from diffusers import DDPMPipeline pipeline = DDPMPipeline.from_pretrained('DeepBird/my_custom_ddpm_train') image = pipeline().images[0] image