requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.0.1
torchvision==0.15.2
transformers==4.33.3
tqdm
decord
timm
einops
opencv_python
torchvision

salesforce-lavis
accelerate