Unverified Commit 427adc89 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Skip `test_contrastive_generate` for `TFXLNet` (#25574)


* fix

* fix

---------

Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent b8f69d0d
No related merge requests found
Showing with 6 additions and 0 deletions
+6 -0
......@@ -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
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment