Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| b9498315db | |||
| cf7b4288af |
+1
-1
@@ -53,7 +53,7 @@ torchaudio==0.8.1
|
||||
torchlibrosa==0.0.9
|
||||
torchvision==0.9.1+cu111
|
||||
tqdm==4.60.0
|
||||
transformers==4.5.1
|
||||
transformers==4.30.0
|
||||
typing-extensions==3.10.0.0
|
||||
urllib3==1.26.5
|
||||
importlib-resources==5.10.0
|
||||
+1
-1
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user