jhj0517 commited on
Commit
b1b9396
·
1 Parent(s): f3f351e

Downgrade torch

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -2,10 +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/cu124
6
 
7
 
8
- torch
 
9
  git+https://github.com/jhj0517/jhj0517-whisper.git
10
  faster-whisper==1.0.3
11
  transformers==4.42.3
 
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==4.42.3