majic31
2024-12-24 23e7ddebccd3b05cf7ef89809bcfe565ad6dfa1f
runtime/python/libtorch/demo_paraformer.py
@@ -1,5 +1,5 @@
from pathlib import Path
from funasr_torch import Paraformer
from funasr_torch.paraformer_bin import Paraformer
model_dir = "iic/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch"
model = Paraformer(model_dir, batch_size=1)  # cpu