Fix pyannote API: use_auth_token -> token
This commit is contained in:
@@ -18,7 +18,7 @@ def get_diarization_pipeline() -> Pipeline:
|
||||
logger.info("Loading diarization model on %s", settings.diarization_device)
|
||||
_diarization_pipeline = Pipeline.from_pretrained(
|
||||
settings.diarization_model,
|
||||
use_auth_token=hf_token,
|
||||
token=hf_token,
|
||||
)
|
||||
_diarization_pipeline.to(torch.device(settings.diarization_device))
|
||||
return _diarization_pipeline
|
||||
|
||||
Reference in New Issue
Block a user