RaynWu2002 commited on
Commit
a466e11
·
verified ·
1 Parent(s): e21f0e6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- PyTorch==2.1.0
2
  numpy==1.23.5
3
  torchvision==0.16.0
4
  scipy==1.11.3
 
1
+ torch==2.1.0
2
  numpy==1.23.5
3
  torchvision==0.16.0
4
  scipy==1.11.3