# 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 is the cu128 build (Blackwell / RTX 5090, CUDA 12.8) via the pytorch cu128 index in the Dockerfile.
torch==2.8.0
transformers>=4.55,<5
trl>=0.23,<0.24
peft
vllm<0.11
datasets>=2.19
huggingface_hub>=0.25
accelerate
