cernenv-trainer / requirements.txt
anugrah55's picture
Pin trl<=0.24.0 + add unsloth_zoo/accelerate/datasets/bitsandbytes
5598d1c verified
raw
history blame contribute delete
980 Bytes
--extra-index-url https://download.pytorch.org/whl/cu124
# Strategy: strict ML pins mirroring requirements-unsloth.txt (the locally
# verified matrix). Loose pins on trl/transformers caused a runtime import
# error: trl resolved to a version requiring transformers>=5, while
# transformers landed on a 4.x release with API trimmed for 5.x, breaking
# `from transformers import PreTrainedModel`.
torch==2.6.0
torchvision==0.21.0
torchaudio==2.6.0
unsloth==2026.4.8
unsloth_zoo
# Pin transformers to 4.x: the 5.x series dropped many legacy model classes
# (e.g. BloomPreTrainedModel) that current peft / trl still reference, which
# broke imports at runtime.
transformers>=4.51.3,<5.0
trl>=0.18.2,<=0.24.0,!=0.19.0
peft>=0.18.0,<0.20
accelerate>=1.0.0
datasets>=2.18.0
bitsandbytes>=0.43.0
xformers
matplotlib>=3.8.0
# Space control panel + env API deps
numpy>=1.24.0
scipy>=1.10.0
pydantic>=2.0.0
fastapi>=0.110.0
uvicorn>=0.27.0
huggingface_hub>=0.24.0
openenv-core[core]>=0.2.3