From 4ace5a95b052d338947fc88809a440ccd55cf6b4 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 16 十一月 2023 16:39:52 +0800
Subject: [PATCH] funasr pages
---
funasr/bin/lm_inference_launch.py | 3 +--
1 files changed, 1 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/lm_inference_launch.py b/funasr/bin/lm_inference_launch.py
index c8482b8..f12f50a 100644
--- a/funasr/bin/lm_inference_launch.py
+++ b/funasr/bin/lm_inference_launch.py
@@ -15,7 +15,6 @@
import numpy as np
import torch
from torch.nn.parallel import data_parallel
-from typeguard import check_argument_types
from funasr.build_utils.build_model_from_file import build_model_from_file
from funasr.build_utils.build_streaming_iterator import build_streaming_iterator
@@ -50,7 +49,6 @@
param_dict: dict = None,
**kwargs,
):
- assert check_argument_types()
ncpu = kwargs.get("ncpu", 1)
torch.set_num_threads(ncpu)
@@ -185,6 +183,7 @@
dtype=dtype,
batch_size=batch_size,
key_file=key_file,
+ preprocess_fn=preprocessor,
num_workers=num_workers,
)
--
Gitblit v1.9.1