From aa340bc2c85f92904b7f86e1a7bb1f838000dc74 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 12 十二月 2022 21:29:52 +0800
Subject: [PATCH] wav
---
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