LTT commited on
Commit
7316a4d
1 Parent(s): 770b62a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -4
requirements.txt CHANGED
@@ -1,7 +1,6 @@
1
- torch==2.1.0
2
- torchvision==0.16.0
3
- torchaudio==2.1.0
4
- -f https://download.pytorch.org/whl/cu118
5
 
6
  pytorch-lightning==2.1.2
7
  gradio==3.41.2
 
1
+ torch==2.1.0+cuda118
2
+ torchvision==0.16.0+cuda118
3
+ torchaudio==2.1.0+cuda118
 
4
 
5
  pytorch-lightning==2.1.2
6
  gradio==3.41.2