requirements.txt

transformers
joblib
accelerate
diffusers
#salesforce-lavis
numpy<2
openai
contexttimer
decord
einops
fairscale
ftfy
iopath
ipython
omegaconf
opencv-python-headless
opendatasets
packaging
pandas
plotly
pre-commit
pycocoevalcap
pycocotools
python-magic
scikit-image
sentencepiece
spacy
timm
torch<=2.4.0
torchvision
tqdm
git+https://github.com/webdataset/webdataset
wheel
git+https://github.com/imageio/imageio
git+https://github.com/matplotlib/matplotlib
#git+https://github.com/pix2pixzero/pix2pix-zero.git