Add model card for Scale-RAE

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +30 -0
README.md ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: transformers
4
+ pipeline_tag: image-to-image
5
+ ---
6
+
7
+ # Scale-RAE: Scaling Text-to-Image Diffusion Transformers with Representation Autoencoders
8
+
9
+ Official model weights for the paper [Scaling Text-to-Image Diffusion Transformers with Representation Autoencoders](https://huggingface.co/papers/2601.16208).
10
+
11
+ Representation Autoencoders (RAEs) enable diffusion modeling in high-dimensional semantic latent spaces. Scale-RAE scales this framework to large-scale, freeform text-to-image generation. RAEs consistently outperform traditional VAEs during pretraining across various model scales, offering faster convergence and better generation quality.
12
+
13
+ - **Project Page:** [https://rae-dit.github.io/scale-rae/](https://rae-dit.github.io/scale-rae/)
14
+ - **GitHub Repository:** [https://github.com/ZitengWangNYU/Scale-RAE](https://github.com/ZitengWangNYU/Scale-RAE)
15
+ - **Paper:** [https://arxiv.org/abs/2601.16208](https://arxiv.org/abs/2601.16208)
16
+
17
+ ## Usage
18
+
19
+ For full text-to-image generation using Scale-RAE, please follow the installation and inference instructions in the [official repository](https://github.com/ZitengWangNYU/Scale-RAE).
20
+
21
+ ## Citation
22
+
23
+ ```bibtex
24
+ @article{scale-rae-2026,
25
+ title={Scaling Text-to-Image Diffusion Transformers with Representation Autoencoders},
26
+ author={Shengbang Tong and Boyang Zheng and Ziteng Wang and Bingda Tang and Nanye Ma and Ellis Brown and Jihan Yang and Rob Fergus and Yann LeCun and Saining Xie},
27
+ journal={arXiv preprint arXiv:2601.16208},
28
+ year={2026}
29
+ }
30
+ ```