anugrah55 commited on
Commit
698bd51
·
verified ·
1 Parent(s): fd802e5

Update CERNenv Space

Browse files
Files changed (1) hide show
  1. space/training/requirements.txt +5 -0
space/training/requirements.txt CHANGED
@@ -7,7 +7,12 @@ torch==2.6.0
7
  torchvision==0.21.0
8
  torchaudio==2.6.0
9
  unsloth==2026.4.8
 
 
 
 
10
  trl>=0.18.2,<=0.24.0,!=0.19.0
 
11
  xformers
12
  matplotlib>=3.8.0
13
  numpy>=1.24.0
 
7
  torchvision==0.21.0
8
  torchaudio==2.6.0
9
  unsloth==2026.4.8
10
+ # Pin transformers to 4.x: the 5.x series dropped many legacy model classes
11
+ # (e.g. BloomPreTrainedModel) that current peft / trl still reference, which
12
+ # broke imports at runtime.
13
+ transformers>=4.51.3,<5.0
14
  trl>=0.18.2,<=0.24.0,!=0.19.0
15
+ peft>=0.18.0,<0.20
16
  xformers
17
  matplotlib>=3.8.0
18
  numpy>=1.24.0