mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-30 03:33:56 +08:00
Update silero.py
This commit is contained in:
@@ -12,13 +12,12 @@ logger = setup_logging()
|
|||||||
class VADProvider(VADProviderBase):
|
class VADProvider(VADProviderBase):
|
||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
logger.bind(tag=TAG).info("SileroVAD", config)
|
logger.bind(tag=TAG).info("SileroVAD", config)
|
||||||
self.model, self.utils = torch.hub.load(
|
self.model, _ = torch.hub.load(
|
||||||
repo_or_dir=config["model_dir"],
|
repo_or_dir=config["model_dir"],
|
||||||
source="local",
|
source="local",
|
||||||
model="silero_vad",
|
model="silero_vad",
|
||||||
force_reload=False,
|
force_reload=False,
|
||||||
)
|
)
|
||||||
get_speech_timestamps, *_ = self.utils
|
|
||||||
|
|
||||||
self.decoder = opuslib_next.Decoder(16000, 1)
|
self.decoder = opuslib_next.Decoder(16000, 1)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user