| # Direct dependencies at the exact versions of the evaluated image (Python 3.12, Linux x86_64, CUDA 12.8). | |
| # requirements.lock is the complete freeze. | |
| torch==2.8.0 | |
| torchvision==0.23.0 | |
| transformers==5.17.0 | |
| tokenizers==0.23.2 | |
| safetensors==0.8.0 | |
| accelerate==1.15.0 | |
| huggingface_hub==1.33.0 | |
| hf-xet==1.6.0 | |
| pillow==12.3.0 | |
| numpy==2.5.3 | |
| flash-linear-attention==0.5.2 | |
| fla-core==0.5.2 | |
| causal-conv1d @ https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.7.0/causal_conv1d-1.7.0%2Bcu12torch2.8cxx11abiTRUE-cp312-cp312-linux_x86_64.whl#sha256=a1fcd5d022b67ddfc9e1c291f890004c1c081a03d66b532f6eebe31219168e0e | |
| triton==3.4.0 | |
| einops==0.8.2 | |
| peft==0.21.0 | |