azalea/CLAP
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix: _batch_generator needs to always yield

Browse Source
This commit is contained in:
Aaron Mihalik
2024-03-10 16:44:30 -04:00
committed by GitHub
parent 1af3d710ee
commit d8b816ef43
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
msclap/CLAPWrapper.py
+1 -1
View File
@@ -312,7 +312,7 @@ class CLAPWrapper():
# batch size is bigger than available audio/text items # batch size is bigger than available audio/text items
if next_batch_idx >= args0_len: if next_batch_idx >= args0_len:
inputs[0] = input_tmp[dataset_idx:] inputs[0] = input_tmp[dataset_idx:]
return func(*tuple(inputs)) yield func(*tuple(inputs))
else: else:
inputs[0] = input_tmp[dataset_idx:next_batch_idx] inputs[0] = input_tmp[dataset_idx:next_batch_idx]
yield func(*tuple(inputs)) yield func(*tuple(inputs))