--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.3.1+cu118
torchvision==0.18.1+cu118
-U pip
wheel
ninja
git+https://github.com/Dao-AILab/flash-attention.git
transformers
gradio
spaces
decord
einops
sentencepiece