Views
No views yet

!python3 -m pip install diffusers==0.21.* accelerate1from matplotlib import pyplot as plt
2from diffusers import DDPMPipeline
3
4ddpm = DDPMPipeline.from_pretrained("mayur7garg/ddpm-fake-icons")
5ddpm.to("cuda")
6
7inference_image = ddpm(
8 batch_size = 36,
9 num_inference_steps = 1000
10).images
11
12plt.figure(figsize = (18, 18), dpi = 120)
13
14for i, image in enumerate(inference_image):
15 plt.subplot(6, 6, i + 1)
16 plt.imshow(image)
17 plt.axis(False)
18plt.tight_layout()
19plt.show()