|
--- |
|
license: apache-2.0 |
|
--- |
|
## Quickstart |
|
```python |
|
!pip install diffusers |
|
from diffusers import DiffusionPipeline |
|
pipeline = DiffusionPipeline.from_pretrained( |
|
"YouLiXiya/yl-consistency", |
|
custom_pipeline="YouLiXiya/yl-consistency", |
|
) |
|
|
|
from PIL import Image |
|
def make_grid(images, rows, cols): |
|
w, h = images[0].size |
|
grid = Image.new('RGB', size=(cols * w, rows * h)) |
|
for i, image in enumerate(images): |
|
grid.paste(image, box=(i % cols * w, i // cols * h)) |
|
return grid |
|
|
|
import matplotlib.pyplot as plt |
|
images = pipeline(batch_size=64, num_inference_steps=10).images |
|
grid = make_grid(images, 8, 8) |
|
plt.imshow(grid) |
|
plt.axis('off') |
|
plt.show() |
|
``` |
