| # Base dependencies |
| torch==2.7.1 |
| torchvision==0.22.1 |
| transformers==4.48.0 |
| datasets==3.1.0 |
| safetensors==0.4.5 |
| Pillow==10.4.0 |
| accelerate==1.3.0 |
| qwen-vl-utils==0.0.14 |
| spaces |
| einops |
|
|
| # Pre-built Wheels (Python 3.12 + CUDA 12 + Torch 2.7) |
| causal-conv1d @ https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.6.0/causal_conv1d-1.6.0+cu12torch2.7cxx11abiTRUE-cp312-cp312-linux_x86_64.whl |
| mamba-ssm @ https://github.com/state-spaces/mamba/releases/download/v2.3.0/mamba_ssm-2.3.0+cu12torch2.7cxx11abiTRUE-cp312-cp312-linux_x86_64.whl |
| flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.7cxx11abiTRUE-cp312-cp312-linux_x86_64.whl |
