Update README.md

README.md

@@ -19,7 +19,40 @@ For more information, please read our [blog post](https://blackforestlabs.ai/ann
 3. Released under the `apache-2.0` licence, the model can be used for personal, scientific, and commercial purposes.
 
 # Usage
-We provide a reference implementation of `FLUX.1 [schnell]`, as well as sampling code, in a dedicated [github repository](https://github.com/black-forest-labs/flux).
+```python
+from diffusers import FluxPipeline
+import torch
+
+ckpt_id = "black-forest-labs/FLUX.1-schnell"
+prompt = 'black forest gateau cake spelling out the words "FLUX SCHNELL", tasty, food photography, dynamic shot'
+height, width = 1024, 1024
+
+pipe = FluxPipeline.from_pretrained(
+    ckpt_id,
+    revision="refs/pr/1",
+    torch_dtype=torch.bfloat16,
+)
+pipe.enable_model_cpu_offload() # uses ~24 GB vram
+# if you don't have >24GB vram, comment the line above and uncomment the lines below; see https://huggingface.co/docs/diffusers/v0.29.2/en/optimization/memory
+#pipe.enable_sequential_cpu_offload()
+#pipe.vae.enable_tiling()
+#pipe.vae.enable_slicing()
+
+images = pipe(
+    prompt,
+    num_inference_steps=1,
+    guidance_scale=0.0,
+    height=height,
+    width=width,
+).images
+
+import matplotlib.pyplot as plt
+for image in images:
+    plt.imshow(image)
+    plt.show()
+```
+
+We also provide a reference implementation of `FLUX.1 [schnell]`, as well as sampling code, in a dedicated [github repository](https://github.com/black-forest-labs/flux).
 Developers and creatives looking to build on top of `FLUX.1 [schnell]` are encouraged to use this as a starting point.
 
 ## API Endpoints