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

Merge pull request #492 from remic33/pyannote3

Browse Source 
Pyannote3
This commit is contained in:
Max Bain
2023-09-26 14:16:57 -07:00
committed by GitHub
parent e94b904308 652aa24919
commit 18abcf46ee
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -19,7 +19,7 @@ setup(
for r in pkg_resources.parse_requirements( for r in pkg_resources.parse_requirements(
open(os.path.join(os.path.dirname(__file__), "requirements.txt")) 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 = { entry_points = {
'console_scripts': ['whisperx=whisperx.transcribe:cli'], 'console_scripts': ['whisperx=whisperx.transcribe:cli'],
}, },

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",
): ):