jhj0517 commited on
Commit
cfb6f77
·
1 Parent(s): 3093cad

Update comment to specify CUDA distribution link

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -1,5 +1,8 @@
1
  # Remove the --extra-index-url line below if you're not using Nvidia GPU.
2
- # If you're using it, Update CUDA version to yours, /cu121 , /cu124 ... etc ( CUDA 12.1 is minimum )
 
 
 
3
  --extra-index-url https://download.pytorch.org/whl/cu121
4
  torch
5
  git+https://github.com/jhj0517/jhj0517-whisper.git
 
1
  # Remove the --extra-index-url line below if you're not using Nvidia GPU.
2
+ # If you're using it, Update CUDA version to yours (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
+
6
  --extra-index-url https://download.pytorch.org/whl/cu121
7
  torch
8
  git+https://github.com/jhj0517/jhj0517-whisper.git