anugrah55 commited on
Commit
027e888
·
verified ·
1 Parent(s): 4fc573b

Update CERNenv Space

Browse files
Files changed (1) hide show
  1. 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
- # Pin a torch + torchao + transformers triple known to work with Unsloth on
3
- # CUDA 12.1. We avoid newer transformers/torchao because they require
4
- # torch>=2.6 (torch.int1 added in 2.6); we pin torch 2.5.1 which is the
5
- # latest stable cu121 wheel that pairs cleanly with Unsloth's kernels.
 
6
  torch==2.5.1
7
  torchvision==0.20.1
8
  torchaudio==2.5.1
9
  torchao==0.7.0
10
- transformers==4.46.3
11
- trl==0.12.2
12
- peft==0.13.2
13
- accelerate==1.1.1
14
- datasets==3.1.0
15
- bitsandbytes==0.44.1
16
- unsloth==2024.12.4
17
- unsloth_zoo==2024.12.7
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