opencv-python
ffmpeg-python
moviepy

pyyaml
json_tricks
yacs
scikit-learn
pandas
timm==0.4.12
numpy==1.23.5
einops
fvcore
transformers==4.19.2
sentencepiece
ftfy
regex
nltk
vision-datasets==0.2.2
pycocotools==2.0.4
diffdist
pyarrow
cityscapesscripts
shapely
scikit-image
mup
scann
kornia==0.6.4
invisible-watermark
diffusers
accelerate
altair
--extra-index-url https://download.pytorch.org/whl/cu113
torch
torchvision==0.13.1+cu113
torchmetrics==0.6.0
openai
git+https://github.com/arogozhnikov/einops.git