| ############################################################################### |
| # CUDA 12.1 runtime + PyTorch 2.1 wheels (official extra index) |
| ############################################################################### |
| --extra-index-url https://download.pytorch.org/whl/cu121 |
|
|
| torch==2.1.0+cu121 # core |
| torchvision==0.16.0+cu121 # must share the same ABI tag |
|
|
| ############################################################################### |
| # Transformers / training utils |
| ############################################################################### |
| transformers==4.46.3 |
| accelerate>=0.26.0 |
| bitsandbytes |
| einops |
| numpy<2.0.0 |
|
|
| ############################################################################### |
| # Video / image processing |
| ############################################################################### |
| decord |
| imageio |
| imageio-ffmpeg |
| opencv-python-headless |
| ffmpeg-python |
| pillow |
|
|
| ############################################################################### |
| # Gradio & web interface |
| ############################################################################### |
| gradio==5.34.0 |
| spaces |
| requests |
|
|
| ############################################################################### |
| # HF ecosystem & model I/O |
| ############################################################################### |
| huggingface_hub |
| sentencepiece |
| safetensors |
|
|
| ############################################################################### |
| # Flash-Attention 2.5.8 ── prebuilt for |
| # • CUDA 12.2 (binary still works under CUDA ≥12.1 drivers*) |
| # • PyTorch 2.1 |
| # • Python 3.10 (cp310) |
| ############################################################################### |
| https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu122torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl |
