Thomas De Decker commited on
Commit
27b1207
1 Parent(s): 7d5d90e

Add PyTorch to requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  orjson==3.6.8
2
- transformers==4.17.0
3
  pandas==1.4.1
4
  numpy==1.22.3
 
1
  orjson==3.6.8
2
+ transformers[torch]==4.17.0
3
  pandas==1.4.1
4
  numpy==1.22.3