| 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") | |
| ``` | |
|  |