av
datasets
einops
hydra-core>=1.3.2
matplotlib
numpy<2
omegaconf>=2.3.0
opencv-python==4.8.0.74
rich>=13.4.2
scikit-learn
seaborn
torch>=2.1.0
torchaudio>=0.13.1
torchvision>=0.16.0
tqdm
transformers
wandb
webdataset@git+https://github.com/elicassion/webdataset.git@elicassion/fix_shuffle_bug