anugrah55 commited on
Commit
ad4b830
·
verified ·
1 Parent(s): d597642

Bump TRL to 0.16.1 (adds GRPOTrainer); transformers 4.51.3, peft 0.14, accelerate 1.4, bnb 0.45.5

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -7
requirements.txt CHANGED
@@ -1,10 +1,11 @@
1
  # Core ML stack. torch is provided by the base image.
2
- transformers==4.46.3
3
- trl==0.13.0
4
- peft==0.13.2
5
- accelerate==1.1.1
6
- bitsandbytes==0.44.1
7
- datasets==3.1.0
 
8
 
9
  # Tokenizer + utility deps
10
  sentencepiece==0.2.0
@@ -14,7 +15,7 @@ safetensors==0.4.5
14
 
15
  # HTTP + Hub
16
  requests==2.32.3
17
- huggingface_hub==0.26.2
18
 
19
  # Misc
20
  numpy==1.26.4
 
1
  # Core ML stack. torch is provided by the base image.
2
+ # TRL >= 0.16 is required for the modern GRPOTrainer / GRPOConfig API.
3
+ transformers==4.51.3
4
+ trl==0.16.1
5
+ peft==0.14.0
6
+ accelerate==1.4.0
7
+ bitsandbytes==0.45.5
8
+ datasets==3.3.2
9
 
10
  # Tokenizer + utility deps
11
  sentencepiece==0.2.0
 
15
 
16
  # HTTP + Hub
17
  requests==2.32.3
18
+ huggingface_hub==0.30.2
19
 
20
  # Misc
21
  numpy==1.26.4