VideoLLaMA2 / requirements.txt
ClownRat's picture
Update torch version.
4105bd7
raw
history blame
614 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118
# basic dependencies
torchvision==0.15.2
torch==2.1.0
transformers==4.37.2
tokenizers==0.15.1
deepspeed==0.13.1
accelerate==0.21.0
peft==0.4.0
timm==0.6.13
numpy
# data processing
decord==0.6.0
imageio==2.34.0
imageio-ffmpeg==0.4.9
pytorchvideo==0.1.5
moviepy==1.0.3
scenedetect==0.6.3
opencv-python==4.7.0.72
# misc
scikit-learn==1.2.2
huggingface_hub==0.22.2
sentencepiece==0.1.99
shortuuid
einops==0.6.1
einops-exts==0.0.4
bitsandbytes==0.41.0
pydantic<2,>=1
markdown2[all]
gradio==3.35.2
gradio_client==0.2.9
requests
httpx==0.24.0
uvicorn
fastapi
wandb