AmirShabani commited on
Commit
0b1da25
·
1 Parent(s): 7cf4040

requirements no torch cpu

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -3,6 +3,6 @@ pillow
3
  numpy==1.23.5
4
  pandas==1.5.3
5
  scikit-learn==1.2.2
6
- torch==2.0.0+cpu
7
- torchvision==0.15.1+cpu
8
- torchaudio==2.0.1+cpu
 
3
  numpy==1.23.5
4
  pandas==1.5.3
5
  scikit-learn==1.2.2
6
+ torch==2.0.0
7
+ torchvision==0.15.1
8
+ torchaudio==2.0.1