Downloads · 30 days
45
16% of all-time downloads
Manojb/taesdxl
taesdxl is a machine learning model from Manojb. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for diffusers. The card lists the license as mit.
TAESDXL is very tiny autoencoder which uses the same "latent API" as SDXL-VAE. TAESDXL is useful for real-time previewing of the SDXL generation process.
Downloads · 30 days
45
16% of all-time downloads
All-time downloads
282
Public
Parameters
2.4M
29.4 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors19.6 MB · 67%
From the Hugging Face model README
TAESDXL is very tiny autoencoder which uses the same "latent API" as SDXL-VAE. TAESDXL is useful for real-time previewing of the SDXL generation process.
This repo contains .safetensors versions of the TAESDXL weights.
For SD1.x / SD2.x, use TAESD instead (the SD and SDXL VAEs are incompatible).
import torch
from diffusers import DiffusionPipeline, AutoencoderTiny
pipe = DiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
)
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "slice of delicious New York-style cheesecake topped with berries, mint, chocolate crumble"
image = pipe(prompt, num_inference_steps=50, generator=torch.Generator("cpu").manual_seed(0x7A35D)).images[0]
image.save("cheesecake_sdxl.png")
