Fix cu_num_generated_tokens slicing logic in LogprobsLists.slice() method

Author: usberkeley

tests/v1/test_outputs.py

@@ -43,7 +43,7 @@ def test_slice_without_cu_num_generated_tokens(self):
         assert sliced.cu_num_generated_tokens is None
 
     def test_slice_from_start(self):
-        """Test slicing from the start position with cu_num_generated_tokens"""
+        """Test slicing from the start position"""
         sliced = self.logprobsLists.slice(0, 2)
         assert len(sliced.logprob_token_ids) == 5
         assert sliced.logprob_token_ids == [
@@ -53,7 +53,7 @@ def test_slice_from_start(self):
         assert sliced.cu_num_generated_tokens == [0, 2, 5]  # Recalculated, starting from 0
 
     def test_slice_from_middle(self):
-        """Test slicing from the middle position with cu_num_generated_tokens"""
+        """Test slicing from the middle position"""
         sliced = self.logprobsLists.slice(1, 3)
         assert len(sliced.logprob_token_ids) == 7
         assert sliced.logprob_token_ids == [