From f80a051337cf98ed939e3e0045fd538eb8e2665b Mon Sep 17 00:00:00 2001 From: Weiguo Zhu Date: Wed, 25 Sep 2024 23:49:11 +0800 Subject: [PATCH] Skip dynamic to static test cases (#9196) --- tests/transformers/test_modeling_common.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/transformers/test_modeling_common.py b/tests/transformers/test_modeling_common.py index 8813c6dee754..51e8745fcb33 100644 --- a/tests/transformers/test_modeling_common.py +++ b/tests/transformers/test_modeling_common.py @@ -926,6 +926,7 @@ def tearDown(self): paddle.disable_static() super().setUp() + @unittest.skip("Paddle enable PIR API in Python") def test_to_static_use_top_k(self): tokenizer = self.TokenizerClass.from_pretrained(self.internal_testing_model) if tokenizer.__class__.__name__ == "LlamaTokenizer": @@ -1006,6 +1007,7 @@ def test_to_static_use_top_k(self): self.assertEqual(len(static_decoded_ids[0]), self.max_new_tokens) self.assertEqual(dygraph_decoded_ids, static_decoded_ids) + @unittest.skip("Paddle enable PIR API in Python") def test_to_static_use_top_p(self): tokenizer = self.TokenizerClass.from_pretrained(self.internal_testing_model) if tokenizer.__class__.__name__ == "LlamaTokenizer":