SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Skip test_contrastive_generate for TFXLNet (#25574)

Browse Source 
* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-08-17 18:56:34 +02:00
committed by GitHub
parent b8f69d0d10
commit 427adc898a
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/xlnet/test_modeling_tf_xlnet.py
View File

@@ -365,6 +365,12 @@ class TFXLNetModelTest(TFModelTesterMixin, PipelineTesterMixin, unittest.TestCas
test_head_masking = False
test_onnx = False
# Note that `TFXLNetModelTest` is not a subclass of `GenerationTesterMixin`, so no contrastive generation tests
# from there is run against `TFXLNetModel`.
@unittest.skip("XLNet has special cache mechanism and is currently not working with contrastive generation")
def test_xla_generate_contrastive(self):
super().test_xla_generate_contrastive()
# TODO: Fix the failed tests
def is_pipeline_test_to_skip(
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name