hackergeek98 commited on
Commit
cb2d3b1
·
verified ·
1 Parent(s): 425ca13

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -6
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
- torch>=2.0.0
2
- transformers>=4.40.0
3
- datasets>=2.18.0
4
- autotrain-advanced>=0.8.0
5
- accelerate>=0.29.0
6
- huggingface_hub>=0.22.0
 
 
1
+ setuptools>=70.0.0
2
+ torch==2.2.1+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
3
+ transformers==4.40.0
4
+ datasets==2.18.0
5
+ autotrain-advanced==0.8.0
6
+ accelerate==0.29.0
7
+ huggingface_hub==0.22.0