# Pinned GPU-worker dependencies baked into the prebuilt AutoSLM worker image.
# MUST stay in sync with autoslm/flash/train.py::WORKER_DEPS (enforced by tests/test_worker_image.py).
# torch 2.10 is the cu128 build (Blackwell / RTX 5090 needs CUDA-13 drivers for PTX JIT;
# see autoslm/flash/train.py::min_cuda_for).
torch==2.10.0
transformers>=5.6,<5.11
trl>=1.5,<1.6
peft>=0.19
vllm==0.19.1
bitsandbytes>=0.49
datasets>=4.7,<6
huggingface_hub>=0.25
accelerate>=1.4
flash-linear-attention==0.5.0
