From 11815a0726f1910eeeba579caf3fd57118e9bd81 Mon Sep 17 00:00:00 2001
From: xuan <admin@exuan.org>
Date: 星期二, 27 二月 2024 10:11:23 +0800
Subject: [PATCH] feat(runtime-html5): display ITN radio (#1392)

---
 funasr/datasets/audio_datasets/preprocessor.py |    9 +++++----
 1 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/funasr/datasets/audio_datasets/preprocessor.py b/funasr/datasets/audio_datasets/preprocessor.py
index a3ba3a5..ab75140 100644
--- a/funasr/datasets/audio_datasets/preprocessor.py
+++ b/funasr/datasets/audio_datasets/preprocessor.py
@@ -26,10 +26,11 @@
 			return waveform
 		speed = random.choice(self.speed_perturb)
 		if speed != 1.0:
-			with torch.no_grad():
-				waveform, _ = torchaudio.sox_effects.apply_effects_tensor(
-					torch.tensor(waveform).view(1, -1), fs, [['speed', str(speed)], ['rate', str(fs)]])
-				waveform = waveform.view(-1)
+			if not isinstance(waveform, torch.Tensor):
+				waveform = torch.tensor(waveform)
+			waveform, _ = torchaudio.sox_effects.apply_effects_tensor(
+				waveform.view(1, -1), fs, [['speed', str(speed)], ['rate', str(fs)]])
+			waveform = waveform.view(-1)
 			
 		return waveform
 

--
Gitblit v1.9.1