Skip dynamic to static test cases (#9196)

PaddlePaddle · Sep 25, 2024 · f80a051 · f80a051
1 parent 1b8f37f
commit f80a051
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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":