From 72ac9e608e24933f030c5afef0bb053873167769 Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Fri, 13 Dec 2024 12:34:14 +0000 Subject: [PATCH] rm pdb Signed-off-by: jiqing-feng --- tests/ipex/test_modeling.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tests/ipex/test_modeling.py b/tests/ipex/test_modeling.py index 85592cbe90..3a6abd9c35 100644 --- a/tests/ipex/test_modeling.py +++ b/tests/ipex/test_modeling.py @@ -377,9 +377,6 @@ def test_compare_with_and_without_past_key_values(self): outputs_model_without_pkv = model_without_pkv.generate( **tokens, min_new_tokens=self.GENERATION_LENGTH, max_new_tokens=self.GENERATION_LENGTH, num_beams=1 ) - import pdb - - pdb.set_trace() self.assertTrue(torch.equal(outputs_model_with_pkv, outputs_model_without_pkv)) 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])