pyDVL
ftfy
regex
tqdm
distributed
zarr
transformers
pycocotools
datasets
diffusers
numpy<2.0.0
# conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=12.1 -c pytorch -c nvidia
# pip install git+https://github.com/openai/CLIP.git
