From 08d39eae7d9886d37ab5d7ae2f579b85b6491e50 Mon Sep 17 00:00:00 2001
From: onlybetheone <iriszhangchong@gmail.com>
Date: 星期四, 22 十二月 2022 17:21:52 +0800
Subject: [PATCH] FunTextProcessing
---
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