From 40b772101a379a9f2ee8dfb2d8278b1dad76420e Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 17 二月 2023 11:44:00 +0800
Subject: [PATCH] Merge pull request #130 from alibaba-damo-academy/dev_lzr
---
funasr/bin/asr_inference_paraformer.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/asr_inference_paraformer.py b/funasr/bin/asr_inference_paraformer.py
index 055a17f..18d788e 100644
--- a/funasr/bin/asr_inference_paraformer.py
+++ b/funasr/bin/asr_inference_paraformer.py
@@ -337,7 +337,7 @@
# assert check_return_type(results)
return results
-class Speech2TextExport(torch.nn.Module):
+class Speech2TextExport:
"""Speech2TextExport class
"""
@@ -416,7 +416,7 @@
self.asr_model = model
@torch.no_grad()
- def forward(
+ def __call__(
self, speech: Union[torch.Tensor, np.ndarray], speech_lengths: Union[torch.Tensor, np.ndarray] = None
):
"""Inference
--
Gitblit v1.9.1