anhvv200053 commited on
Commit
0ecbab7
·
verified ·
1 Parent(s): d3436d9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -7
requirements.txt CHANGED
@@ -1,12 +1,8 @@
1
- huggingface_hub==0.22.2
2
  transformers==4.33.2
3
-
4
- torch>=2.0.1
5
- datasets>=2.14.5
6
- bitsandbytes==0.41.1
7
- accelerate==0.21.0
8
  sentence-transformers
9
- peft>=0.4.0
10
  trl
11
  rouge-score
12
  sentencepiece
 
 
1
  transformers==4.33.2
2
+ torch # Không cần phải chỉ định phiên bản cụ thể nếu bạn chỉ cần hỗ trợ CPU
3
+ datasets
 
 
 
4
  sentence-transformers
5
+ peft
6
  trl
7
  rouge-score
8
  sentencepiece