Views
No views yet
1from diffusers import DiffusionPipeline
2
3pipe = DiffusionPipeline.from_pretrained("maxmarcon/klimt-diffusion").to("cuda")
4
5images = pipe(10)
6
7import matplotlib.pyplot as plt
8
9fig, axes = plt.subplots(2, 5, figsize=(15, 6))
10for axis, image in zip(axes.flat, images[0]):
11 axis.imshow(image)
12 axis.axis("off")
13
14plt.tight_layout()
15plt.show()