Pytorch implementation for the paper titled "SimpleAR: Pushing the Frontier of Autoregressive Visual Generation"
Python
431
11 commits
updated Jun 20, 2025
This paper presents SimpleAR, a vanilla autoregressive visual generation model that achieves state-of-the-art text-to-image generation performance. First the first time, we demonstrate that:
We open-sourced all the training and inference code, hoping to show the potential of autoregressive visual generation and encourage more participation in this research field.
For basic usage (pretraining, SFT, inference without vLLM), you can install the dependencies with:
python3 -m venv env
source env/bin/activate
pip install -e ".[train]"
While for advanced usage, please refer to TRAIN.md (GRPO training) and EVAL.md (inference with vLLM) to setup the environments, respectively.
We provide both SFT and RL checkpoints:
| name | GenEval | DPG | HF weights 🤗 |
|---|---|---|---|
| SimpleAR-0.5B-SFT | 0.53 | 79.34 | simplear-0.5B-sft |
| SimpleAR-0.5B-RL | 0.59 | 79.66 | simplear-0.5B-grpo |
| SimpleAR-1.5B-SFT | 0.61 | 80.11 | simplear-1.5B-sft |
| SimpleAR-1.5B-RL | 0.63 | 81.31 | simplear-1.5B-grpo |
Cosmos is used as our visual tokenizer, you can download and put it under ./checkpoints/:
cd checkpoints
git lfs install
git clone https://huggingface.co/nvidia/Cosmos-1.0-Tokenizer-DV8x16x16
You can directly load SimpleAR with from_pretrained now 🤗! We provide the demo code in PLAY.md.
Please find the instructions on data preparation and training here.
We provide scripts to evaluate our released checkpoints on GenEval and DPG-Bench. Please see EVAL.md for more details.
Also, you can generate images with SimpleAR using generate.py. We implement different acceleration approaches, e.g., vLLM, speculative jacobi decoding. Please refer to EVAL.md.
1024 x 1024 generation results by SimpleAR.
If you find this repository helpful, please consider citing:
@article{wang2025simplear,
title={SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RL},
author={Wang, Junke and Tian, Zhi and Wang, Xun and Zhang, Xinyu and Huang, Weilin and Wu, Zuxuan and Jiang, Yu-Gang},
journal={arXiv preprint arXiv:2504.11455},
year={2025}
}
We thank Peize Sun, Rui Tian, Feng Li, and Teng Yao for their valuable discussions.
830 followers · starred May 2025
123 followers · starred Apr 2025
9 followers · starred Apr 2025
12 followers · starred Apr 2025
Pytorch implementation for the paper titled "SimpleAR: Pushing the Frontier of Autoregressive Visual Generation"
Python
431
11 commits
updated Jun 20, 2025
This paper presents SimpleAR, a vanilla autoregressive visual generation model that achieves state-of-the-art text-to-image generation performance. First the first time, we demonstrate that:
We open-sourced all the training and inference code, hoping to show the potential of autoregressive visual generation and encourage more participation in this research field.
For basic usage (pretraining, SFT, inference without vLLM), you can install the dependencies with:
python3 -m venv env
source env/bin/activate
pip install -e ".[train]"
While for advanced usage, please refer to TRAIN.md (GRPO training) and EVAL.md (inference with vLLM) to setup the environments, respectively.
We provide both SFT and RL checkpoints:
| name | GenEval | DPG | HF weights 🤗 |
|---|---|---|---|
| SimpleAR-0.5B-SFT | 0.53 | 79.34 | simplear-0.5B-sft |
| SimpleAR-0.5B-RL | 0.59 | 79.66 | simplear-0.5B-grpo |
| SimpleAR-1.5B-SFT | 0.61 | 80.11 | simplear-1.5B-sft |
| SimpleAR-1.5B-RL | 0.63 | 81.31 | simplear-1.5B-grpo |
Cosmos is used as our visual tokenizer, you can download and put it under ./checkpoints/:
cd checkpoints
git lfs install
git clone https://huggingface.co/nvidia/Cosmos-1.0-Tokenizer-DV8x16x16
You can directly load SimpleAR with from_pretrained now 🤗! We provide the demo code in PLAY.md.
Please find the instructions on data preparation and training here.
We provide scripts to evaluate our released checkpoints on GenEval and DPG-Bench. Please see EVAL.md for more details.
Also, you can generate images with SimpleAR using generate.py. We implement different acceleration approaches, e.g., vLLM, speculative jacobi decoding. Please refer to EVAL.md.
1024 x 1024 generation results by SimpleAR.
If you find this repository helpful, please consider citing:
@article{wang2025simplear,
title={SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RL},
author={Wang, Junke and Tian, Zhi and Wang, Xun and Zhang, Xinyu and Huang, Weilin and Wu, Zuxuan and Jiang, Yu-Gang},
journal={arXiv preprint arXiv:2504.11455},
year={2025}
}
We thank Peize Sun, Rui Tian, Feng Li, and Teng Yao for their valuable discussions.
830 followers · starred May 2025
123 followers · starred Apr 2025
9 followers · starred Apr 2025
12 followers · starred Apr 2025