nielsr HF Staff commited on
Commit
ebaffa9
·
verified ·
1 Parent(s): a5d6b86

Add model card and metadata

Browse files

Hi! I'm Niels from the Hugging Face community science team.

This PR adds a comprehensive model card to the repository. It includes:
- Metadata for `license`, `library_name`, and `pipeline_tag`.
- Links to the paper ([Scaling Text-to-Image Diffusion Transformers with Representation Autoencoders](https://huggingface.co/papers/2601.16208)), project page, and official GitHub repository.
- A brief description of the model and its findings.
- Sample CLI usage for inference as found in the official documentation.
- Citation information.

This update will make the model more discoverable and provide users with the necessary context and links to the official implementation.

Files changed (1) hide show
  1. README.md +33 -0
README.md ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: transformers
4
+ pipeline_tag: image-to-image
5
+ ---
6
+
7
+ # Scaling Text-to-Image Diffusion Transformers with Representation Autoencoders (Scale-RAE)
8
+
9
+ This repository contains artifacts related to the paper [Scaling Text-to-Image Diffusion Transformers with Representation Autoencoders](https://huggingface.co/papers/2601.16208).
10
+
11
+ - **Project Page:** [https://rae-dit.github.io/scale-rae/](https://rae-dit.github.io/scale-rae/)
12
+ - **GitHub Repository:** [https://github.com/ZitengWangNYU/Scale-RAE](https://github.com/ZitengWangNYU/Scale-RAE)
13
+
14
+ ## Introduction
15
+ Representation Autoencoders (RAEs) provide a simplified and powerful alternative to VAEs for large-scale text-to-image generation. Scale-RAE demonstrates that training diffusion models in high-dimensional semantic latent spaces (using encoders like SigLIP-2) leads to faster convergence, better generation quality, and improved stability compared to state-of-the-art VAE-based foundations.
16
+
17
+ ## Usage
18
+ For detailed instructions on installation, training, and inference, please visit the [official GitHub repository](https://github.com/ZitengWangNYU/Scale-RAE).
19
+
20
+ The implementation supports GPU inference and TPU training. To generate images with pre-trained models:
21
+ ```bash
22
+ python cli.py t2i --prompt "Can you generate a photo of a cat on a windowsill?"
23
+ ```
24
+
25
+ ## Citation
26
+ ```bibtex
27
+ @article{scale-rae-2026,
28
+ title={Scaling Text-to-Image Diffusion Transformers with Representation Autoencoders},
29
+ 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},
30
+ journal={arXiv preprint arXiv:2601.16208},
31
+ year={2026}
32
+ }
33
+ ```