turtle170 commited on
Commit
753f05b
·
verified ·
1 Parent(s): dc01d21

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -4
requirements.txt CHANGED
@@ -1,16 +1,18 @@
1
 
2
  transformers>=4.48.0
3
- trl>=0.12.0
4
  peft>=0.14.0
5
  accelerate>=1.2.0
6
  bitsandbytes>=0.45.0
7
 
8
- flash-attn --no-build-isolation
9
 
10
  unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
11
- unsloth-zoo
12
 
13
  psutil
14
  tensorboard
15
  sentencepiece
16
- protobuf
 
 
 
1
 
2
  transformers>=4.48.0
3
+ trl>=0.13.0
4
  peft>=0.14.0
5
  accelerate>=1.2.0
6
  bitsandbytes>=0.45.0
7
 
8
+ flash-attn --no-build-isolation
9
 
10
  unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
11
+ unsloth-zoo>=2026.1.3
12
 
13
  psutil
14
  tensorboard
15
  sentencepiece
16
+ protobuf
17
+ torchao
18
+ cut-cross-entropy