---
library_name: llamagen
tags:
- image-generation
- image-tokenizer
- imagenet
- vq-vae
---

# VAE checkpoints for ImageNet LlamaGen GPT-XL

This repository contains the two VQ-16 class-conditional ImageNet tokenizers used by
the paired LlamaGen GPT-XL experiment:

- `base_vae_vq_ds16_c2i.pt`: official FoundationVision/LlamaGen VQ-16 c2i checkpoint.
- `reward_vae_vq_ds16_c2i_seed3101_step1000.pt`: reward-tuned VQ-16 c2i checkpoint,
  exported from the seed 3101, step 1000 tokenizer run.

Both checkpoints use a 16,384-entry codebook and produce 24 x 24 = 576 tokens for a
384px ImageNet input. They are intended for the ImageNet c2i training pipeline in
https://github.com/hzeroyuke/vae-posttraining.

## SHA256

| File | SHA256 |
| --- | --- |
| `base_vae_vq_ds16_c2i.pt` | `109aa8afb2cf3761eec23cdc8644154cb498f5ab7eef2a35264d25e5e0499f7d` |
| `reward_vae_vq_ds16_c2i_seed3101_step1000.pt` | `55c06b0ab914f9f40b24eba07033f7c452eecb6233562baf27e69121282434ea` |

The checkpoints are distributed for research use. Follow the licenses and terms of
LlamaGen, ImageNet, PyTorch, and any downstream evaluation dependencies.