Update CERNenv Space

space/training/requirements.txt

@@ -1,21 +1,14 @@
---extra-index-url https://download.pytorch.org/whl/cu121
-# Pinned set known to work with Unsloth + GRPO on CUDA 12.1.
-# Critical constraints:
-#   * torch < 2.6 (torch.int1 lives in 2.6+; older torchao breaks otherwise)
-#   * trl >= 0.14 (GRPOConfig / GRPOTrainer were added in 0.14)
-#   * transformers compatible with both unsloth and trl 0.15
-torch==2.5.1
-torchvision==0.20.1
-torchaudio==2.5.1
-torchao==0.7.0
-transformers==4.48.3
-trl==0.15.2
-peft==0.14.0
-accelerate==1.3.0
-datasets==3.2.0
-bitsandbytes==0.45.0
-unsloth==2025.2.4
-unsloth_zoo==2025.2.3
+--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
+trl>=0.18.2,<=0.24.0,!=0.19.0
+xformers
 matplotlib>=3.8.0
 numpy>=1.24.0
 scipy>=1.10.0