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