Improve model card and add metadata

README.md

@@ -1,9 +1,58 @@
 ---
 license: apache-2.0
+pipeline_tag: unconditional-image-generation
+tags:
+- image-generation
+- autoregressive
 ---
 
-Website: https://bar-gen.github.io/
+# Autoregressive Image Generation with Masked Bit Modeling (BAR)
 
-Code: https://github.com/amazon-far/BAR
+This repository contains the checkpoints for **BAR** (*masked **B**it **A**uto**R**egressive modeling*), a scalable framework for discrete visual generation that challenges the dominance of continuous pipelines.
 
-Paper:
+[**Project Page**](https://bar-gen.github.io/) | [**Paper (ArXiv)**](https://huggingface.co/papers/2602.09024) | [**GitHub**](https://github.com/amazon-far/BAR)
+
+## Introduction
+
+Contrary to the belief that discrete tokenizers are intrinsically inferior, BAR demonstrates that the performance gap arises primarily from the total number of bits allocated in the latent space. By equipping an autoregressive transformer with a **Masked Bit Modeling (MBM)** head, BAR predicts discrete tokens through progressively generating their constituent bits.
+
+- **Scalability**: Supports arbitrary codebook sizes with reduced memory complexity.
+- **Performance**: Achieves a state-of-the-art gFID of **0.99** on ImageNet-256.
+- **Efficiency**: Offers faster convergence and significantly lower sampling costs compared to diffusion models.
+
+## Model Zoo
+
+| Model | Config | Size | gFID | IS |
+| ------------- | ------------- | ------------- | ------------- | ------------- |
+| BAR-B/4 | [bar_b_patch4.yaml](https://github.com/amazon-far/BAR/blob/main/configs/generator/bar_b_patch4.yaml) | 416M | 2.34 | 274.7 |
+| BAR-B/2 | [bar_b_patch2.yaml](https://github.com/amazon-far/BAR/blob/main/configs/generator/bar_b_patch2.yaml) | 415M | 1.35 | 293.4 |
+| BAR-B | [bar_b.yaml](https://github.com/amazon-far/BAR/blob/main/configs/generator/bar_b.yaml) | 415M | 1.13 | 289.0 |
+| BAR-L | [bar_l.yaml](https://github.com/amazon-far/BAR/blob/main/configs/generator/bar_l.yaml) | 1.1B | 0.99 | 296.9 |
+| BAR-L-res512 | [bar_l_res512.yaml](https://github.com/amazon-far/BAR/blob/main/configs/generator/bar_l_res512.yaml) | 1.1B | 1.09 | 311.1 |
+
+## Sample Usage
+
+To generate samples using the weights in this repository, please follow the setup instructions in the [official GitHub repository](https://github.com/amazon-far/BAR). Below is an example command for generating samples with **BAR-L**:
+
+```bash
+torchrun --nnodes=1 --nproc_per_node=1 --rdzv-endpoint=localhost:9999 \
+    sample_imagenet.py \
+    config=configs/generator/bar_l.yaml \
+    experiment.output_dir="bar_l" \
+    experiment.generator_checkpoint=assets/generator/bar_l.bin \
+    experiment.tokenizer_checkpoint=assets/tokenizer/bar_fsq_16bits_ft.bin \
+    model.generator.guidance_scale=5.3 \
+    model.generator.mbm_head.randomize_temperature=3.0 \
+    'model.generator.mbm_head.tokens_allocation=[2,2,5,7]'
+```
+
+## Citation
+
+```bibtex
+@article{yu2026autoregressive,
+  title     = {Autoregressive Image Generation with Masked Bit Modeling},
+  author    = {Yu, Qihang and Liu, Qihao and He, Ju and Zhang, Xinyang and Liu, Yang and Chen, Liang-Chieh and Chen, Xi},
+  journal   = {arXiv preprint},
+  year      = {2026}
+}
+```