File size: 690 Bytes
07389d5
36dab63
a026fa0
8ccff75
07389d5
 
 
8ccff75
a3cb9f2
d78080d
2abf781
7a5e0df
ae7032c
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# 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