hpoghos commited on
Commit
307f3e5
·
verified ·
1 Parent(s): bc9ad47

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu117
2
- torch
3
- torchdata
4
- torchvision
5
  accelerate==0.28.0
6
  bitsandbytes==0.43.0
7
  transformers==4.39.3
 
1
  --extra-index-url https://download.pytorch.org/whl/cu117
2
+ torch==2.0.0+cu117
3
+ torchdata==0.6.0+cu117
4
+ torchvision==0.15.1+cu117
5
  accelerate==0.28.0
6
  bitsandbytes==0.43.0
7
  transformers==4.39.3