From b17908473daab16846dfd97be73b6d0d4ff1cc78 Mon Sep 17 00:00:00 2001
From: Remc <caland.remi@gmail.com>
Date: Tue, 26 Sep 2023 17:18:20 +0200
Subject: [PATCH] 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",
     ):