video-llama2-test / requirements.txt
comidan's picture
Update requirements.txt
5b8e709
--extra-index-url https://download.pytorch.org/whl/cu113
torch==1.12.1
torchvision==0.13.1
torchaudio==0.12.1
transformers==4.28.0
tqdm
decord
timm
einops
opencv_python
torchvision
pytorchvideo
salesforce-lavis
accelerate