Use pyannote develop branch and torch version 2

sorgfresser · May 2, 2023 · 0671892 · 0671892
1 parent 5becc99
commit 0671892
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,5 @@
-torch==1.13.1
-torchaudio==0.13.1
-pyannote.audio==2.1.1
+torch==2.0.0
+torchaudio==2.0.1
 faster-whisper
 transformers
 ffmpeg-python==0.2.0

diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
         for r in pkg_resources.parse_requirements(
             open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
         )
-    ],
+    ] + ["pyannote.audio @ https://github.com/pyannote/pyannote-audio/archive/refs/heads/develop.zip"],
     entry_points = {
         'console_scripts': ['whisperx=whisperx.transcribe:cli'],
     },