|
--- |
|
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] |
|
``` |
