From ab653d3871f72f7f6cd1ac3126b3df722f4c7943 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期五, 20 十月 2023 15:33:09 +0800
Subject: [PATCH] add rwkv encoder
---
funasr/bin/asr_infer.py | 7 ++++---
1 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/funasr/bin/asr_infer.py b/funasr/bin/asr_infer.py
index ef3ed6e..4648fb3 100644
--- a/funasr/bin/asr_infer.py
+++ b/funasr/bin/asr_infer.py
@@ -1922,9 +1922,6 @@
):
from funasr.tasks.whisper import ASRTask
- from funasr.utils.whisper_utils.transcribe import transcribe
- from funasr.utils.whisper_utils.audio import pad_or_trim, log_mel_spectrogram
- from funasr.utils.whisper_utils.decoding import DecodingOptions, detect_language, decode
# 1. Build ASR model
scorers = {}
@@ -1987,6 +1984,10 @@
"""
+ from funasr.utils.whisper_utils.transcribe import transcribe
+ from funasr.utils.whisper_utils.audio import pad_or_trim, log_mel_spectrogram
+ from funasr.utils.whisper_utils.decoding import DecodingOptions, detect_language, decode
+
speech = speech[0]
speech = pad_or_trim(speech)
mel = log_mel_spectrogram(speech).to(self.device)
--
Gitblit v1.9.1