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