File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -926,6 +926,7 @@ def tearDown(self):
926
926
paddle .disable_static ()
927
927
super ().setUp ()
928
928
929
+ @unittest .skip ("Paddle enable PIR API in Python" )
929
930
def test_to_static_use_top_k (self ):
930
931
tokenizer = self .TokenizerClass .from_pretrained (self .internal_testing_model )
931
932
if tokenizer .__class__ .__name__ == "LlamaTokenizer" :
@@ -1006,6 +1007,7 @@ def test_to_static_use_top_k(self):
1006
1007
self .assertEqual (len (static_decoded_ids [0 ]), self .max_new_tokens )
1007
1008
self .assertEqual (dygraph_decoded_ids , static_decoded_ids )
1008
1009
1010
+ @unittest .skip ("Paddle enable PIR API in Python" )
1009
1011
def test_to_static_use_top_p (self ):
1010
1012
tokenizer = self .TokenizerClass .from_pretrained (self .internal_testing_model )
1011
1013
if tokenizer .__class__ .__name__ == "LlamaTokenizer" :
You can’t perform that action at this time.
0 commit comments