Rishi Desai commited on
Commit
d68e3ce
·
1 Parent(s): 4cbae4e

HF complaining about torch

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu124
2
- torch==2.4.0
3
- torchaudio==2.4.0
4
- torchvision==0.19.0
5
  aiofiles==24.1.0
6
  aiohappyeyeballs==2.6.1
7
  aiohttp==3.11.16
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ torch==2.1.2
3
+ torchaudio==2.1.2
4
+ torchvision==0.16.2
5
  aiofiles==24.1.0
6
  aiohappyeyeballs==2.6.1
7
  aiohttp==3.11.16