Koala-video-llm / requirements.txt
Reuben Tan
initial commit
b2afdba
--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