yl-consistency / README.md
YouLiXiya's picture
Update README.md
5934e65
---
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()
```
<div>
<a href="https://colab.research.google.com/github/Youlixiya/ylcm/blob/master/scripts/unconditional_cm_cifar10_32_pipeline.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>
</div>