Update README.md

README.md

@@ -1,4 +1,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).to("cuda")
+
+images = pipeline(num_inference_steps=40).images
+images[0]
+```