Rajeev-86 commited on
Commit
cb713ee
·
1 Parent(s): da96e9f

changed the pytorch version in requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- torch==2.4.0+cu118
2
- torchvision==0.23.0+cu118
 
3
  numpy==2.0.2
4
  pillow==11.3.0
 
1
+ --index-url https://download.pytorch.org/whl/cu126
2
+ torch==2.8.0+cu126
3
+ torchvision==0.23.0+cu126
4
  numpy==2.0.2
5
  pillow==11.3.0