VideoLLaMA2 / requirements.txt
ClownRat's picture
Fix dependency.
ff899bc
raw
history blame
558 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118
huggingface_hub==0.22.2
torch>=2.0.1
torchvision>=0.15.2
transformers==4.37.2
tokenizers==0.15.1
sentencepiece==0.1.99
shortuuid
deepspeed==0.13.1
accelerate==0.21.0
peft==0.4.0
decord==0.6.0
pytorchvideo==0.1.5
imageio==2.34.0
imageio-ffmpeg==0.4.9
moviepy==1.0.3
scenedetect==0.6.3
opencv-python
numpy
scikit-learn==1.2.2
einops==0.6.1
einops-exts==0.0.4
timm==0.6.13
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