From b17908473daab16846dfd97be73b6d0d4ff1cc78 Mon Sep 17 00:00:00 2001 From: Remc Date: Tue, 26 Sep 2023 17:18:20 +0200 Subject: [PATCH 1/2] correct 3.0 pyannote weights --- whisperx/diarize.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/whisperx/diarize.py b/whisperx/diarize.py index e50dc0f..d59a847 100644 --- a/whisperx/diarize.py +++ b/whisperx/diarize.py @@ -6,10 +6,11 @@ import torch from .audio import load_audio, SAMPLE_RATE + class DiarizationPipeline: def __init__( self, - model_name="pyannote/speaker-diarization@2.1", + model_name="pyannote/speaker-diarization-3.0", use_auth_token=None, device: Optional[Union[str, torch.device]] = "cpu", ): From 652aa24919812c5afee48691f9d93ca79b8f9dad Mon Sep 17 00:00:00 2001 From: Remc Date: Tue, 26 Sep 2023 23:04:28 +0200 Subject: [PATCH 2/2] change pyannote version --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 1e1e6f5..ae72640 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ setup( for r in pkg_resources.parse_requirements( open(os.path.join(os.path.dirname(__file__), "requirements.txt")) ) - ] + ["pyannote.audio @ git+https://github.com/pyannote/pyannote-audio@11b56a137a578db9335efc00298f6ec1932e6317"], + ] + ["pyannote.audio @ git+https://github.com/pyannote/pyannote-audio@db24eb6c60a26804b1f07a6c2b39055716beb852"], entry_points = { 'console_scripts': ['whisperx=whisperx.transcribe:cli'], },