File size: 997 Bytes
2d55237 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
---
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")
```
 |