fix the last ipex failing test_compare_with_and_without_past_key_values

Author: IlyasMoutawwakil

tests/ipex/test_modeling.py

@@ -357,13 +357,8 @@ def test_compare_with_and_without_past_key_values(self):
                 **tokens, min_new_tokens=self.GENERATION_LENGTH, max_new_tokens=self.GENERATION_LENGTH, num_beams=1
             )
         self.assertTrue(torch.equal(outputs_model_with_pkv, outputs_model_without_pkv))
-        self.assertEqual(outputs_model_with_pkv.shape[1], self.GENERATION_LENGTH + tokens.shape[1])
-        self.assertEqual(outputs_model_without_pkv.shape[1], self.GENERATION_LENGTH + tokens.shape[1])
-        # self.assertTrue(
-        #     without_pkv_timer.elapsed / with_pkv_timer.elapsed > self.SPEEDUP_CACHE,
-        #     f"With pkv latency: {with_pkv_timer.elapsed:.3f} ms, without pkv latency: {without_pkv_timer.elapsed:.3f} ms,"
-        #     f" speedup: {without_pkv_timer.elapsed / with_pkv_timer.elapsed:.3f}",
-        # )
+        self.assertEqual(outputs_model_with_pkv.shape[1], self.GENERATION_LENGTH + tokens.input_ids.shape[1])
+        self.assertEqual(outputs_model_without_pkv.shape[1], self.GENERATION_LENGTH + tokens.input_ids.shape[1])
 
 
 class IPEXModelForAudioClassificationTest(unittest.TestCase):