from diffusers import DDPMPipeline pipeline = DDPMPipeline.from_pretrained('ZYJ-3721/sd-butterflies-64').to('cuda') image = pipeline(num_inference_steps=50).images[0] image