Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformers

About

We introduce Sana, a text-to-image framework that can efficiently generate images up to 4096$\times$4096 resolution. Sana can synthesize high-resolution, high-quality images with strong text-image alignment at a remarkably fast speed, deployable on laptop GPU. Core designs include: (1) Deep compression autoencoder: unlike traditional AEs, which compress images only 8$\times$, we trained an AE that can compress images 32$\times$, effectively reducing the number of latent tokens. (2) Linear DiT: we replace all vanilla attention in DiT with linear attention, which is more efficient at high resolutions without sacrificing quality. (3) Decoder-only text encoder: we replaced T5 with modern decoder-only small LLM as the text encoder and designed complex human instruction with in-context learning to enhance the image-text alignment. (4) Efficient training and sampling: we propose Flow-DPM-Solver to reduce sampling steps, with efficient caption labeling and selection to accelerate convergence. As a result, Sana-0.6B is very competitive with modern giant diffusion model (e.g. Flux-12B), being 20 times smaller and 100+ times faster in measured throughput. Moreover, Sana-0.6B can be deployed on a 16GB laptop GPU, taking less than 1 second to generate a 1024$\times$1024 resolution image. Sana enables content creation at low cost. Code and model will be publicly released.

Enze Xie, Junsong Chen, Junyu Chen, Han Cai, Haotian Tang, Yujun Lin, Zhekai Zhang, Muyang Li, Ligeng Zhu, Yao Lu, Song Han• 2024

Related benchmarks

TaskDatasetResultRank
Text-to-Image GenerationGenEval
Overall Score66
467
Class-conditional Image GenerationImageNet 256x256
Inception Score (IS)341.9
441
Class-conditional Image GenerationImageNet 256x256 (val)
FID2.32
293
Text-to-Image GenerationGenEval
GenEval Score66
277
Text-to-Image GenerationDPG-Bench
Overall Score84.8
173
Text-to-Image GenerationGenEval (test)
Two Obj. Acc77
169
Text-to-Image GenerationMS-COCO 2014 (val)--
128
Text-to-Image GenerationDPG-Bench
DPG Score84.8
89
Text-to-Image GenerationGenEval
Two Objects77
87
Image GenerationImageNet-1K 256x256 (val)
Inception Score48.15
85
Showing 10 of 20 rows

Other info

Follow for update