Skip to content

Commit 72ac9e6

Browse files
committed
rm pdb
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent bce9aa9 commit 72ac9e6

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/ipex/test_modeling.py

-3
Original file line numberDiff line numberDiff line change
@@ -377,9 +377,6 @@ def test_compare_with_and_without_past_key_values(self):
377377
outputs_model_without_pkv = model_without_pkv.generate(
378378
**tokens, min_new_tokens=self.GENERATION_LENGTH, max_new_tokens=self.GENERATION_LENGTH, num_beams=1
379379
)
380-
import pdb
381-
382-
pdb.set_trace()
383380
self.assertTrue(torch.equal(outputs_model_with_pkv, outputs_model_without_pkv))
384381
self.assertEqual(outputs_model_with_pkv.shape[1], self.GENERATION_LENGTH + tokens.input_ids.shape[1])
385382
self.assertEqual(outputs_model_without_pkv.shape[1], self.GENERATION_LENGTH + tokens.input_ids.shape[1])

0 commit comments

Comments
 (0)