peakji commited on
Commit
7bd4097
·
1 Parent(s): 762029c

Enable GPU acceleration

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -1
requirements.txt CHANGED
@@ -1,2 +1,7 @@
 
 
 
 
 
1
  torch
2
- transformers
 
1
+ accelerate
2
+ datasets
3
+ huggingface_hub
4
+ matplotlib
5
+ sentencepiece
6
  torch
7
+ transformers