shounakpaul95 commited on
Commit
131fb67
1 Parent(s): ede242e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  seaborn
2
  scipy
3
- pytorch
4
- pytorch-cuda
5
  datasets==2.14.5
6
  gradio
7
  huggingface-hub==0.18.0
 
1
  seaborn
2
  scipy
3
+ pytorch==2
4
+ pytorch-cuda==12.1
5
  datasets==2.14.5
6
  gradio
7
  huggingface-hub==0.18.0