README.md

metadata

datasets:
  - huggan/few-shot-aurora

---

datasets: - huggan/few-shot-aurora

Have you ever seen aurora with your own eyes? Check the picture I got in Alaska in Winter. Beautiful right?

Usage

Option 1 (Slow)

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('li-yan/diffusion-aurora-256')
image = pipeline().images[0]
image

Option 2 (Fast)

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]