Merge pull request #30 from amihalik/patch-1

bugfix: _batch_inference needs to always `yield`
2024-04-04 10:58:47 -07:00
parent 1af3d710ee d8b816ef43
commit 59bc8446e3
1 changed files with 1 additions and 1 deletions
@@ -312,7 +312,7 @@ class CLAPWrapper():
            # batch size is bigger than available audio/text items
            if next_batch_idx >= args0_len:
                inputs[0] = input_tmp[dataset_idx:]
-                return func(*tuple(inputs))
+                yield func(*tuple(inputs))
            else:
                inputs[0] = input_tmp[dataset_idx:next_batch_idx]
                yield func(*tuple(inputs))