jhj0517 commited on
Commit
0e3bef2
·
1 Parent(s): f458c92

Update torch according to CTranslate2

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -2,11 +2,11 @@
2
  # If you're using it, update url to your CUDA version (CUDA 12.1 is minimum requirement):
3
  # For CUDA 12.1, use : https://download.pytorch.org/whl/cu121
4
  # For CUDA 12.4, use : https://download.pytorch.org/whl/cu124
5
- --extra-index-url https://download.pytorch.org/whl/cu121
6
 
7
 
8
- torch==2.3.1
9
- torchaudio==2.3.1
10
  git+https://github.com/jhj0517/jhj0517-whisper.git
11
  faster-whisper==1.0.3
12
  transformers
 
2
  # If you're using it, update url to your CUDA version (CUDA 12.1 is minimum requirement):
3
  # For CUDA 12.1, use : https://download.pytorch.org/whl/cu121
4
  # For CUDA 12.4, use : https://download.pytorch.org/whl/cu124
5
+ --extra-index-url https://download.pytorch.org/whl/cu124
6
 
7
 
8
+ torch
9
+ torchaudio
10
  git+https://github.com/jhj0517/jhj0517-whisper.git
11
  faster-whisper==1.0.3
12
  transformers