Add ngram_size option

This commit is contained in:
litagin02
2024-03-14 17:42:29 +09:00
parent 8c95f0ff75
commit 6f42a1514f
2 changed files with 18 additions and 2 deletions

View File

@@ -165,6 +165,14 @@ if __name__ == "__main__":
logger.warning(f"Output directory {output_dir} already exists, deleting...")
shutil.rmtree(output_dir)
# モデルをダウンロードしておく
_ = torch.hub.load(
repo_or_dir="litagin02/silero-vad",
model="silero_vad",
onnx=True,
trust_repo=True,
)
# Silero VADのモデルは、同じインスタンスで並列処理するとおかしくなるらしい
# ワーカーごとにモデルをロードするようにするため、Queueを使って処理する
def process_queue(