rivapereira123 commited on
Commit
93b6b74
·
verified ·
1 Parent(s): 7f15d74

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,10 +1,10 @@
1
  # Core ML and AI
2
  torch>=2.0.0
3
- transformers==4.40.1
 
4
  sentence-transformers>=2.6.1
5
  accelerate==0.24.1
6
  peft==0.10.0
7
- trl==0.7.10
8
  datasets
9
 
10
  # Optional: 8-bit quantization (uncomment if needed)
 
1
  # Core ML and AI
2
  torch>=2.0.0
3
+ transformers==4.36.2
4
+ trl==0.7.10
5
  sentence-transformers>=2.6.1
6
  accelerate==0.24.1
7
  peft==0.10.0
 
8
  datasets
9
 
10
  # Optional: 8-bit quantization (uncomment if needed)