yusendai peakji commited on
Commit
9d89cf4
·
1 Parent(s): 762029c

Enable GPU acceleration (#3)

Browse files

- Enable GPU acceleration (7bd4097f8074974e683dbde2cb126e5cb47cc70b)


Co-authored-by: Yichao 'Peak' Ji <peakji@users.noreply.huggingface.co>

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