Spaces:
Sleeping
Sleeping
Update CERNenv Space
Browse files- space/training/requirements.txt +13 -12
space/training/requirements.txt
CHANGED
|
@@ -1,20 +1,21 @@
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 2 |
-
#
|
| 3 |
-
#
|
| 4 |
-
# torch
|
| 5 |
-
#
|
|
|
|
| 6 |
torch==2.5.1
|
| 7 |
torchvision==0.20.1
|
| 8 |
torchaudio==2.5.1
|
| 9 |
torchao==0.7.0
|
| 10 |
-
transformers==4.
|
| 11 |
-
trl==0.
|
| 12 |
-
peft==0.
|
| 13 |
-
accelerate==1.
|
| 14 |
-
datasets==3.
|
| 15 |
-
bitsandbytes==0.
|
| 16 |
-
unsloth==
|
| 17 |
-
unsloth_zoo==
|
| 18 |
matplotlib>=3.8.0
|
| 19 |
numpy>=1.24.0
|
| 20 |
scipy>=1.10.0
|
|
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 2 |
+
# Pinned set known to work with Unsloth + GRPO on CUDA 12.1.
|
| 3 |
+
# Critical constraints:
|
| 4 |
+
# * torch < 2.6 (torch.int1 lives in 2.6+; older torchao breaks otherwise)
|
| 5 |
+
# * trl >= 0.14 (GRPOConfig / GRPOTrainer were added in 0.14)
|
| 6 |
+
# * transformers compatible with both unsloth and trl 0.15
|
| 7 |
torch==2.5.1
|
| 8 |
torchvision==0.20.1
|
| 9 |
torchaudio==2.5.1
|
| 10 |
torchao==0.7.0
|
| 11 |
+
transformers==4.48.3
|
| 12 |
+
trl==0.15.2
|
| 13 |
+
peft==0.14.0
|
| 14 |
+
accelerate==1.3.0
|
| 15 |
+
datasets==3.2.0
|
| 16 |
+
bitsandbytes==0.45.0
|
| 17 |
+
unsloth==2025.2.4
|
| 18 |
+
unsloth_zoo==2025.2.3
|
| 19 |
matplotlib>=3.8.0
|
| 20 |
numpy>=1.24.0
|
| 21 |
scipy>=1.10.0
|