| --- |
| 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. |
|
|
