From d6fdd1c79364668afbf171bd18586dd1d7570d20 Mon Sep 17 00:00:00 2001
From: 凌匀 <ailsa.zly@alibaba-inc.com>
Date: 星期四, 16 二月 2023 14:56:32 +0800
Subject: [PATCH] support vad streaming decoder
---
funasr/tasks/vad.py | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/funasr/tasks/vad.py b/funasr/tasks/vad.py
index dfd07c4..e2a9123 100644
--- a/funasr/tasks/vad.py
+++ b/funasr/tasks/vad.py
@@ -235,7 +235,7 @@
cls, args: argparse.Namespace, train: bool
) -> Optional[Callable[[str, Dict[str, np.array]], Dict[str, np.ndarray]]]:
assert check_argument_types()
- #if args.use_preprocessor:
+ # if args.use_preprocessor:
# retval = CommonPreprocessor(
# train=train,
# # NOTE(kamo): Check attribute existence for backward compatibility
@@ -254,7 +254,7 @@
# if hasattr(args, "rir_scp")
# else None,
# )
- #else:
+ # else:
# retval = None
retval = None
assert check_return_type(retval)
@@ -291,7 +291,8 @@
model_class = model_choices.get_class(args.model)
except AttributeError:
model_class = model_choices.get_class("e2evad")
- model = model_class(encoder=encoder, vad_post_args=args.vad_post_conf)
+ model = model_class(encoder=encoder, vad_post_args=args.vad_post_conf,
+ streaming=args.encoder_conf.get('streaming', False))
return model
--
Gitblit v1.9.1