From 83072c5f1525c7c88070006559d7c15eb308be79 Mon Sep 17 00:00:00 2001 From: drryanhuang Date: Mon, 30 Dec 2024 11:10:42 +0000 Subject: [PATCH] fix stupid device variable --- audio/audiotools/core/audio_signal.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/audio/audiotools/core/audio_signal.py b/audio/audiotools/core/audio_signal.py index f50172b0d..80a4130d6 100644 --- a/audio/audiotools/core/audio_signal.py +++ b/audio/audiotools/core/audio_signal.py @@ -787,14 +787,15 @@ class AudioSignal( if device is None or "" == device: return self elif 'cpu' == device: - device = paddle.to_tensor( + self.audio_data = paddle.to_tensor( self.audio_data, place=paddle.CPUPlace()) elif 'gpu' == device or 'cuda' == device: - device = paddle.to_tensor( + self.audio_data = paddle.to_tensor( self.audio_data, place=paddle.CUDAPlace()) - device = device.replace("cuda", - "gpu") if "cuda" in device else device - self.audio_data = self.audio_data.to(device) + else: + device = device.replace("cuda", + "gpu") if "cuda" in device else device + self.audio_data = self.audio_data.to(device) return self def float(self):