Views
No views yet
1!pip install diffusers
2from diffusers import DiffusionPipeline
3pipeline = DiffusionPipeline.from_pretrained(
4 "YouLiXiya/yl-consistency",
5 custom_pipeline="YouLiXiya/yl-consistency",
6)
7
8from PIL import Image
9def make_grid(images, rows, cols):
10 w, h = images[0].size
11 grid = Image.new('RGB', size=(cols * w, rows * h))
12 for i, image in enumerate(images):
13 grid.paste(image, box=(i % cols * w, i // cols * h))
14 return grid
15
16import matplotlib.pyplot as plt
17images = pipeline(batch_size=64, num_inference_steps=10).images
18grid = make_grid(images, 8, 8)
19plt.imshow(grid)
20plt.axis('off')
21plt.show()