madebyollin/taesd3

🤗 Hugging Face sourcemit2M params10 MBsafetensors✓ 2 checksumsupdated today

Submit in one command

Run it next to your model folder. It makes the torrent, checks your files against Hugging Face, and submits it. You just start seeding and paste your key from your account. It only reads your files and never changes them. Read the script first if you like.

curl -fsSL https://pirateface.co/package.sh | bash -s -- --repo madebyollin/taesd3 ./model-folder

Needs a seeder →

🍰 Tiny AutoEncoder for Stable Diffusion 3

TAESD3 is very tiny autoencoder which uses the same "latent API" as Stable Diffusion 3's VAE. TAESD3 is useful for real-time previewing of the SD3 generation process.

This repo contains .safetensors versions of the TAESD3 weights.

Using in 🧨 diffusers

import torch
from diffusers import StableDiffusion3Pipeline, AutoencoderTiny

pipe = StableDiffusion3Pipeline.from_pretrained(
    "stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16
)
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesd3", torch_dtype=torch.float16)
pipe.vae.config.shift_factor = 0.0
pipe = pipe.to("cuda")

prompt = "slice of delicious New York-style berry cheesecake"
image = pipe(prompt, num_inference_steps=25).images[0]
image.save("cheesecake.png")