Update README.md

README.md

@@ -1,3 +1,35 @@
----
-license: mit
----
+---
+license: mit
+---
+
+# 🍰 Tiny AutoEncoder for FLUX.1
+
+[TAEF1](https://github.com/madebyollin/taesd) is very tiny autoencoder which uses the same "latent API" as FLUX.1's VAE.
+FLUX.1 is useful for real-time previewing of the FLUX.1 generation process.
+
+This repo contains `.safetensors` versions of the TAEF1 weights.
+
+## Using in 🧨 diffusers
+
+```python
+import torch
+from diffusers import FluxPipeline, AutoencoderTiny
+
+pipe = FluxPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16
+)
+pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.bfloat16)
+pipe.enable_sequential_cpu_offload()
+
+prompt = "slice of delicious New York-style berry cheesecake"
+image = pipe(
+    prompt,
+    guidance_scale=0.0,
+    num_inference_steps=4,
+    max_sequence_length=256,
+).images[0]
+image.save("cheesecake.png")
+```
+
+
+![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/630447d40547362a22a969a2/5FbkYmKc4JsULnskQmzLe.jpeg)