Update requirements.txt
Browse files- requirements.txt +5 -4
requirements.txt
CHANGED
|
@@ -1,8 +1,8 @@
|
|
| 1 |
-
torch
|
|
|
|
| 2 |
omegaconf
|
| 3 |
-
torchaudio
|
| 4 |
einops
|
| 5 |
-
numpy
|
| 6 |
transformers
|
| 7 |
sentencepiece
|
| 8 |
tqdm
|
|
@@ -10,4 +10,5 @@ tensorboard
|
|
| 10 |
descript-audiotools>=0.7.2
|
| 11 |
descript-audio-codec
|
| 12 |
scipy==1.10.1
|
| 13 |
-
huggingface-hub==0.25.2
|
|
|
|
|
|
| 1 |
+
torch==2.2.0
|
| 2 |
+
torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu118
|
| 3 |
omegaconf
|
|
|
|
| 4 |
einops
|
| 5 |
+
numpy<2
|
| 6 |
transformers
|
| 7 |
sentencepiece
|
| 8 |
tqdm
|
|
|
|
| 10 |
descript-audiotools>=0.7.2
|
| 11 |
descript-audio-codec
|
| 12 |
scipy==1.10.1
|
| 13 |
+
huggingface-hub==0.25.2
|
| 14 |
+
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|