Pin trl<=0.24.0 + add unsloth_zoo/accelerate/datasets/bitsandbytes

requirements.txt

@@ -1,24 +1,31 @@
---extra-index-url https://download.pytorch.org/whl/cu124
-# Strategy: pin only torch (so we get the right CUDA wheel) and unsloth (which
-# locks the rest of the matrix — trl, transformers, peft, etc — transitively
-# via its package metadata). We avoid hand-pinning the surrounding libraries
-# because hand-pins kept producing import-time syntax errors from skew.
-torch==2.6.0
-torchvision==0.21.0
-torchaudio==2.6.0
-unsloth==2026.4.8
-# 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
-xformers
-matplotlib>=3.8.0
-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
+--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