RamAnanth1 commited on
Commit
22fffda
·
verified ·
1 Parent(s): 784c43e

Update requirements.txt

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