Encode to SLAT step hanging

[cs-mshah]

Hi,
I'm trying to prepare the dataset and run the training. The python dataset_toolkits/encode_latent.py --output_dir datasets/toys4k_tiny is hanging and doesn't proceed.