dangtiendatdat commited on
Commit
d0f6a7d
·
verified ·
1 Parent(s): b772bbd

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -2
requirements.txt CHANGED
@@ -1,8 +1,7 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu121
2
  --find-links https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.4.0_cu121.html
3
 
4
- torch==2.4.0
5
- torchvision==0.19.0
6
  numpy==1.23.5
7
  transformers
8
  datasets
 
1
  --extra-index-url https://download.pytorch.org/whl/cu121
2
  --find-links https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.4.0_cu121.html
3
 
4
+ torch
 
5
  numpy==1.23.5
6
  transformers
7
  datasets