guowenxiang commited on
Commit
bee9452
1 Parent(s): d7702c4

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -1
requirements.txt CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  torchlibrosa==0.1.0
2
  pytorch-lightning==2.3.0
3
  jupyter==1.0.0
@@ -17,4 +21,5 @@ torchdyn==1.0.6
17
  ninja==1.11.1.1
18
  protobuf==5.27.1
19
  openai==1.34.0
20
- flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch2.1cxx11abiFALSE-cp39-cp39-linux_x86_64.whl
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.1.0
3
+ torchvision==0.16.0
4
+ torchaudio==2.1.0
5
  torchlibrosa==0.1.0
6
  pytorch-lightning==2.3.0
7
  jupyter==1.0.0
 
21
  ninja==1.11.1.1
22
  protobuf==5.27.1
23
  openai==1.34.0
24
+ numpy==1.23.5
25
+ flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl