output_dir = None

游雁

2023-05-22 d16e9d8248e0691e7c4eb43c9cc7eba3aed27270

 funasr/runtime/python/utils/test_rtf.py

@@ -14,9 +14,9 @@
args = parser.parse_args()


from funasr.runtime.python.libtorch.torch_paraformer import Paraformer
from funasr.runtime.python.libtorch.funasr_torch import Paraformer
if args.backend == "onnx":
   from funasr.runtime.python.onnxruntime.rapid_paraformer import Paraformer
   from funasr.runtime.python.onnxruntime.funasr_onnx import Paraformer
   
model = Paraformer(args.model_dir, batch_size=1, quantize=args.quantize, intra_op_num_threads=args.intra_op_num_threads)