wu0792
2024-07-25 0d36a5e176224123c826a254e80c61b7b6dc435a
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