mirror of
https://github.com/mudler/LocalAI.git
synced 2026-06-18 13:49:09 -04:00
fix(whisperx): use whisperx.diarize.DiarizationPipeline with token kwarg (#10389)
Signed-off-by: Anai-Guo <antai12232931@outlook.com>
This commit is contained in:
@@ -79,6 +79,7 @@ class BackendServicer(backend_pb2_grpc.BackendServicer):
|
||||
|
||||
def AudioTranscription(self, request, context):
|
||||
import whisperx
|
||||
from whisperx.diarize import DiarizationPipeline
|
||||
|
||||
resultSegments = []
|
||||
text = ""
|
||||
@@ -106,8 +107,8 @@ class BackendServicer(backend_pb2_grpc.BackendServicer):
|
||||
# Diarize if requested and HF token is available
|
||||
if request.diarize and self.hf_token:
|
||||
if self.diarize_pipeline is None:
|
||||
self.diarize_pipeline = whisperx.DiarizationPipeline(
|
||||
use_auth_token=self.hf_token,
|
||||
self.diarize_pipeline = DiarizationPipeline(
|
||||
token=self.hf_token,
|
||||
device=self.device,
|
||||
)
|
||||
diarize_segments = self.diarize_pipeline(audio)
|
||||
|
||||
Reference in New Issue
Block a user