| --- |
| license: mit |
| --- |
| |
| # 🍰 Tiny AutoEncoder for FLUX.1 |
|
|
| [TAEF1](https://github.com/elarasoft/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("smartguy0505/tae", 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") |
| ``` |
|
|
|
|
|  |
