From 22f0bd826565dcdeba78771deb32d1e92d500e44 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 12 十二月 2022 21:32:36 +0800
Subject: [PATCH] Merge pull request #14 from alibaba-damo-academy/dev
---
funasr/bin/asr_inference_paraformer.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/funasr/bin/asr_inference_paraformer.py b/funasr/bin/asr_inference_paraformer.py
index 9dcd0b8..5673403 100755
--- a/funasr/bin/asr_inference_paraformer.py
+++ b/funasr/bin/asr_inference_paraformer.py
@@ -197,9 +197,9 @@
# data: (Nsamples,) -> (1, Nsamples)
# lengths: (1,)
- if len(speech.size()) < 3:
- speech = speech.unsqueeze(0).to(getattr(torch, self.dtype))
- speech_lengths = speech.new_full([1], dtype=torch.long, fill_value=speech.size(1))
+ # if len(speech.size()) < 3:
+ # speech = speech.unsqueeze(0).to(getattr(torch, self.dtype))
+ # speech_lengths = speech.new_full([1], dtype=torch.long, fill_value=speech.size(1))
lfr_factor = max(1, (speech.size()[-1]//80)-1)
batch = {"speech": speech, "speech_lengths": speech_lengths}
--
Gitblit v1.9.1