Install silero via pip

This commit is contained in:
Lewis Wolfgang
2024-08-05 10:01:27 -04:00
parent 65b136bf15
commit 97cfbfee1d
2 changed files with 3 additions and 18 deletions

View File

@@ -15,6 +15,7 @@ from pipecat.vad.vad_analyzer import VADAnalyzer, VADParams, VADState
from loguru import logger
try:
from silero_vad import load_silero_vad
import torch
# We don't use torchaudio here, but we need to try importing it because
# Silero uses it.
@@ -37,10 +38,6 @@ class SileroVADAnalyzer(VADAnalyzer):
self,
*,
sample_rate: int = 16000,
version: str = "v5.0",
force_reload: bool = False,
skip_validation: bool = True,
trust_repo: bool = True,
params: VADParams = VADParams()):
super().__init__(sample_rate=sample_rate, num_channels=1, params=params)
@@ -49,11 +46,7 @@ class SileroVADAnalyzer(VADAnalyzer):
logger.debug("Loading Silero VAD model...")
(self._model, _) = torch.hub.load(repo_or_dir=f"snakers4/silero-vad:{version}",
model="silero_vad",
force_reload=force_reload,
skip_validation=skip_validation,
trust_repo=trust_repo)
self._model = load_silero_vad()
self._last_reset_time = 0
@@ -94,20 +87,12 @@ class SileroVAD(FrameProcessor):
self,
*,
sample_rate: int = 16000,
version: str = "v5.0",
force_reload: bool = False,
skip_validation: bool = True,
trust_repo: bool = True,
vad_params: VADParams = VADParams(),
audio_passthrough: bool = False):
super().__init__()
self._vad_analyzer = SileroVADAnalyzer(
sample_rate=sample_rate,
version=version,
force_reload=force_reload,
skip_validation=skip_validation,
trust_repo=trust_repo,
params=vad_params)
self._audio_passthrough = audio_passthrough