shenyunhang commited on
Commit
f5abbdf
·
verified ·
1 Parent(s): 8f2612a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -42,8 +42,9 @@ onnxruntime
42
  modelscope
43
  word2number
44
 
45
- torch
46
- torchaudio
 
47
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
48
 
49
 
 
42
  modelscope
43
  word2number
44
 
45
+ torch==2.6.0
46
+ torchvision==0.21.0
47
+ torchaudio==2.6.0
48
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
49
 
50