RamAnanth1 commited on
Commit
827efd9
·
verified ·
1 Parent(s): eaf3e02

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -5,5 +5,5 @@ scipy
5
  accelerate
6
  torchvision
7
  matplotlib
8
- torch==2.4.1
9
  --extra-index-url https://download.pytorch.org/whl/cu118
 
5
  accelerate
6
  torchvision
7
  matplotlib
8
+ torch==2.6
9
  --extra-index-url https://download.pytorch.org/whl/cu118