shenoy commited on
Commit
e2da97b
1 Parent(s): 8fd1ef9

Fixing dependencies

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -2,4 +2,5 @@ accelerate==0.21.0
2
  peft==0.4.0
3
  bitsandbytes==0.40.2
4
  transformers==4.30.2
5
- trl==0.4.7
 
 
2
  peft==0.4.0
3
  bitsandbytes==0.40.2
4
  transformers==4.30.2
5
+ trl==0.4.7
6
+ scikit-learn