diff --git a/msclap/CLAPWrapper.py b/msclap/CLAPWrapper.py
index b75eb37..095a8e9 100644
--- a/msclap/CLAPWrapper.py
+++ b/msclap/CLAPWrapper.py
@@ -312,7 +312,7 @@ def _generic_batch_inference(self, func, *args):
             # 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))