feat: pass model to 3.1 in code

This commit is contained in:
Remc
2023-11-17 11:12:16 +01:00
parent 1d7f8ccbf1
commit 20161935a1

View File

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