VideoLLaMA2 / requirements.txt
ClownRat's picture
Update requirements.
2c2d48f
--extra-index-url https://download.pytorch.org/whl/cu117
# basic dependencies
torchvision==0.15.2
torch==2.0.1
transformers==4.37.2
tokenizers==0.15.1
deepspeed==0.13.1
accelerate
timm
numpy
spaces
peft==0.4.0
# 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