This repository provides the ImageNet-1K dataset pre-encoded with the Stable Diffusion XL VAE encoder and quantized to uint8, allowing for faster training of latent diffusion models by eliminating the need for on-the-fly encoding.
Key Features
Reduces quantization error by 2dB PSNR compared to a linear encoding scheme
Provided in both 256 and 512 resolutions
Compatible with NumPy, JAX, and PyTorch
Usage
Loading the dataset… See the full description on the dataset page: https://huggingface.co/datasets/jon-kyl/imagenet-sdxl-quantized.