Spaces:
Sleeping
Sleeping
Update CERNenv Space
Browse files
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
|