From cf7b4288aff929679fd1fc75a0261568bb7ba84a Mon Sep 17 00:00:00 2001 From: Soham Date: Thu, 13 Apr 2023 11:53:23 -0700 Subject: [PATCH] call cuda function --- src/CLAPWrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/CLAPWrapper.py b/src/CLAPWrapper.py index d69d3bf..7db3622 100644 --- a/src/CLAPWrapper.py +++ b/src/CLAPWrapper.py @@ -149,7 +149,7 @@ class CLAPWrapper(): audio_tensor = self.load_audio_into_tensor( audio_file, self.args.duration, resample) audio_tensor = audio_tensor.reshape( - 1, -1).cuda if self.use_cuda and torch.cuda.is_available() else audio_tensor.reshape(1, -1) + 1, -1).cuda() if self.use_cuda and torch.cuda.is_available() else audio_tensor.reshape(1, -1) audio_tensors.append(audio_tensor) return self.default_collate(audio_tensors)