Removed sequence_lengths to avoid creating batches in a slow way