From 20161935a1348c66ee1de2ee27d01f8388bcfc11 Mon Sep 17 00:00:00 2001
From: Remc <caland.remi@gmail.com>
Date: Fri, 17 Nov 2023 11:12:16 +0100
Subject: [PATCH] feat: pass model to 3.1 in code

---
 whisperx/diarize.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/whisperx/diarize.py b/whisperx/diarize.py
index 55fe220..c1e30be 100644
--- a/whisperx/diarize.py
+++ b/whisperx/diarize.py
@@ -10,7 +10,7 @@ from .audio import load_audio, SAMPLE_RATE
 class DiarizationPipeline:
     def __init__(
         self,
-        model_name="pyannote/speaker-diarization-3.0",
+        model_name="pyannote/speaker-diarization-3.1",
         use_auth_token=None,
         device: Optional[Union[str, torch.device]] = "cpu",
     ):