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 #97 from smly/gpu-vad-filter

Browse Source 
GPU acceleration when using VAD filters
This commit is contained in:
m-bain
2023-02-21 18:57:14 +00:00
committed by GitHub
parent f7093e60d3 57f5957e0e
commit 2b1ffa12b8
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
whisperx/transcribe.py
View File

@ -645,9 +645,12 @@ def cli():
if hf_token is None: if hf_token is None:
print("Warning, no huggingface token used, needs to be saved in environment variable, otherwise will throw error loading VAD model...") print("Warning, no huggingface token used, needs to be saved in environment variable, otherwise will throw error loading VAD model...")
from pyannote.audio import Inference from pyannote.audio import Inference
vad_pipeline = Inference("pyannote/segmentation", vad_pipeline = Inference(
pre_aggregation_hook=lambda segmentation: segmentation, "pyannote/segmentation",
use_auth_token=hf_token) pre_aggregation_hook=lambda segmentation: segmentation,
use_auth_token=hf_token,
device=torch.device(device),
)
diarize_pipeline = None diarize_pipeline = None
if diarize: if diarize: