From 6bdae5723e403584ee3507675b3e4e922a34f246 Mon Sep 17 00:00:00 2001
From: lingyunfly <121302812+lingyunfly@users.noreply.github.com>
Date: 星期五, 09 六月 2023 16:19:35 +0800
Subject: [PATCH] Optimize memory (#612)
---
funasr/bin/sv_inference_launch.py | 3 +--
1 files changed, 1 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/sv_inference_launch.py b/funasr/bin/sv_inference_launch.py
index 8e00730..dbddd9f 100755
--- a/funasr/bin/sv_inference_launch.py
+++ b/funasr/bin/sv_inference_launch.py
@@ -34,7 +34,6 @@
from funasr.utils.cli_utils import get_commandline_args
from funasr.tasks.sv import SVTask
-from funasr.tasks.asr import ASRTask
from funasr.torch_utils.device_funcs import to_device
from funasr.torch_utils.set_all_random_seed import set_all_random_seed
from funasr.utils import config_argparse
@@ -115,7 +114,7 @@
data_path_and_name_and_type = [raw_inputs, "speech", "waveform"]
# 3. Build data-iterator
- loader = ASRTask.build_streaming_iterator(
+ loader = SVTask.build_streaming_iterator(
data_path_and_name_and_type,
dtype=dtype,
batch_size=batch_size,
--
Gitblit v1.9.1