correct 3.0 pyannote weights

2023-09-26 17:18:20 +02:00
parent e94b904308
commit b17908473d
1 changed files with 2 additions and 1 deletions
--- 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",
    ):