chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

correct 3.0 pyannote weights

Browse Source
This commit is contained in:
Remc
2023-09-26 17:18:20 +02:00
parent e94b904308
commit b17908473d
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisperx/diarize.py
View File

@ -6,10 +6,11 @@ import torch
from .audio import load_audio, SAMPLE_RATE from .audio import load_audio, SAMPLE_RATE
class DiarizationPipeline: class DiarizationPipeline:
def __init__( def __init__(
self, self,
model_name="pyannote/speaker-diarization@2.1", model_name="pyannote/speaker-diarization-3.0",
use_auth_token=None, use_auth_token=None,
device: Optional[Union[str, torch.device]] = "cpu", device: Optional[Union[str, torch.device]] = "cpu",
): ):