roychao19477 commited on
Commit
fa25b56
·
1 Parent(s): f515c0f

Upload wheel

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -2
requirements.txt CHANGED
@@ -8,7 +8,6 @@ pesq
8
  einops
9
  flash_attn==2.5.7
10
  numpy==1.26.4
11
- torch==2.1
12
- torchaudio==2.1
13
  numpy==1.26.4
14
  numpy==1.26.4
 
8
  einops
9
  flash_attn==2.5.7
10
  numpy==1.26.4
11
+ torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118
 
12
  numpy==1.26.4
13
  numpy==1.26.4