requirements.txt

aiohttp==3.8.4
aiosignal==1.3.1
async-timeout==4.0.2
attrs==22.2.0
chardet==5.1.0
contourpy==1.0.7
cycler==0.11.0
filelock==3.9.0
fonttools==4.38.0
frozenlist==1.3.3
huggingface-hub
importlib-resources==5.12.0
kiwisolver==1.4.4
matplotlib==3.7.0
multidict==6.0.4
packaging==23.0
psutil==5.9.4
pycocotools==2.0.6
pyparsing==3.0.9
python-dateutil==2.8.2
pyyaml==6.0
regex==2022.10.31
tqdm==4.64.1
timm==0.9.2
spacy==3.5.1
webdataset==0.2.48
scikit-learn
scipy
yarl==1.8.2
zipp==3.14.0
omegaconf==2.3.0
iopath==0.1.10
decord==0.6.0
tenacity==8.2.2
pycocoevalcap
sentence-transformers
umap-learn
notebook
gradio==3.24.1
gradio-client==0.0.8
torch>=2.0
torchvision>=0.15.0
transformers==4.37.2
pyiqa
xformers
accelerate>=0.20.3
peft==0.6.2
tokenizers
lightning>=2.0.2
open_clip_torch
rouge
pytorch-fid
torch-fidelity
torchmetrics
diffusers==0.14.0
opencv-python-headless
prettytable
deepspeed>=0.9.3
datasets
# flash-attn --no-build-isolation