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)
|
logger.info("Loading diarization model on %s", settings.diarization_device)
|
||||||
_diarization_pipeline = Pipeline.from_pretrained(
|
_diarization_pipeline = Pipeline.from_pretrained(
|
||||||
settings.diarization_model,
|
settings.diarization_model,
|
||||||
use_auth_token=hf_token,
|
token=hf_token,
|
||||||
)
|
)
|
||||||
_diarization_pipeline.to(torch.device(settings.diarization_device))
|
_diarization_pipeline.to(torch.device(settings.diarization_device))
|
||||||
return _diarization_pipeline
|
return _diarization_pipeline
|
||||||
|
|||||||
Reference in New Issue
Block a user