wangjin2000 commited on
Commit
db69cc7
·
verified ·
1 Parent(s): afd04ce

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -3,7 +3,7 @@
3
  # platform: linux-64
4
 
5
  pytorch
6
- pytorch-cuda==11.8
7
  accelerate==0.27.2
8
  scikit-learn==1.3.0
9
  numpy==1.24.3
 
3
  # platform: linux-64
4
 
5
  pytorch
6
+ pytorch-cuda
7
  accelerate==0.27.2
8
  scikit-learn==1.3.0
9
  numpy==1.24.3