File size: 645 Bytes
0e76abd 94b73c1 4fd02c8 94b73c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
---
datasets:
- huggan/few-shot-aurora
---
---
datasets:
- huggan/few-shot-aurora
---
# Usage
## Option 1 (Slow)
```python
from diffusers import DDPMPipeline
pipeline = DDPMPipeline.from_pretrained('li-yan/diffusion-aurora-256')
image = pipeline().images[0]
image
```
## Option 2 (Fast)
```python
from diffusers import DiffusionPipeline, DDIMScheduler
scheduler = DDIMScheduler.from_pretrained('li-yan/diffusion-aurora-256')
scheduler.set_timesteps(num_inference_steps=40)
pipeline = DiffusionPipeline.from_pretrained(
'li-yan/diffusion-aurora-256', scheduler=scheduler)
images = pipeline(num_inference_steps=40).images
images[0]
``` |