[本地唤醒模型]:完成KWS模型下载和检查,包含manifest、配置和model-check
This commit is contained in:
@@ -18,7 +18,7 @@ from .stt import MetadataSttProvider, SherpaOnnxSttProvider
|
||||
from .transport import MemoryAudioTransport, sounddevice_device_report
|
||||
from .tts import SineTtsProvider
|
||||
from .vad import EnergyVadProvider, SherpaOnnxVadProvider, VadRecorder
|
||||
from .wakeword import KeywordWakeWordProvider
|
||||
from .wakeword import KeywordWakeWordProvider, SherpaOnnxKeywordWakeWordProvider
|
||||
|
||||
|
||||
def main(argv: list[str] | None = None) -> int:
|
||||
@@ -53,6 +53,10 @@ def main(argv: list[str] | None = None) -> int:
|
||||
"llm_stream": config.llm_stream,
|
||||
"llm_api_key_present": bool(config.llm_api_key),
|
||||
"asset_dir": str(config.asset_dir),
|
||||
"wake_provider": config.wake_provider,
|
||||
"wake_keywords_file": str(config.wake_keywords_file) if config.wake_keywords_file else "",
|
||||
"wake_kws_threshold": config.wake_kws_threshold,
|
||||
"wake_kws_score": config.wake_kws_score,
|
||||
"speech_provider": config.speech_provider,
|
||||
"asr_model": config.asr_model,
|
||||
"tts_model": config.tts_model,
|
||||
@@ -83,6 +87,13 @@ def main(argv: list[str] | None = None) -> int:
|
||||
provider_load_checked = False
|
||||
if not errors:
|
||||
try:
|
||||
SherpaOnnxKeywordWakeWordProvider(
|
||||
models_dir,
|
||||
keyword=config.wake_word,
|
||||
keywords_file=config.wake_keywords_file,
|
||||
threshold=config.wake_kws_threshold,
|
||||
score=config.wake_kws_score,
|
||||
).load()
|
||||
SherpaOnnxVadProvider(models_dir).load()
|
||||
SherpaOnnxSttProvider(str(models_dir)).load()
|
||||
provider_load_checked = True
|
||||
@@ -129,6 +140,10 @@ def main(argv: list[str] | None = None) -> int:
|
||||
audio_output_device=config.audio_output_device,
|
||||
asset_dir=config.asset_dir,
|
||||
log_dir=config.log_dir,
|
||||
wake_provider=config.wake_provider,
|
||||
wake_keywords_file=config.wake_keywords_file,
|
||||
wake_kws_threshold=config.wake_kws_threshold,
|
||||
wake_kws_score=config.wake_kws_score,
|
||||
speech_provider=config.speech_provider,
|
||||
asr_model=config.asr_model,
|
||||
tts_model=config.tts_model,
|
||||
|
||||
Reference in New Issue
Block a user