[!WARNING]
Better latent: I advise you to use another dataset
https://huggingface.co/datasets/cloneofsimo/imagenet.int8 which is already compressed (5Go only) and use a better latent model (SDXL)
This dataset is the latent representation of the imagenet dataset using the stability VAE stabilityai/sd-vae-ft-ema.
Every image_latent is of shape (4, 32, 32).
If you want to retrieve the original image you have to use the model used to create the latent image :
vae_model =… See the full description on the dataset page:
https://huggingface.co/datasets/Forbu14/imagenet-1k-latent.
