Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 7d87122493 |
+2
-2
@@ -38,8 +38,8 @@ requests==2.25.1
|
||||
resampy==0.2.2
|
||||
sacremoses==0.0.45
|
||||
scikit-learn==0.24.2
|
||||
scipy==1.10.0
|
||||
sentry-sdk==1.0.0
|
||||
scipy==1.6.3
|
||||
sentry-sdk==1.14.0
|
||||
shortuuid==1.0.1
|
||||
six==1.15.0
|
||||
smmap==4.0.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