diff --git a/tests/transformers/tests/models/mistral/test_modeling_mistral.py b/tests/transformers/tests/models/mistral/test_modeling_mistral.py index d6a0bb3333..962eea1b0e 100644 --- a/tests/transformers/tests/models/mistral/test_modeling_mistral.py +++ b/tests/transformers/tests/models/mistral/test_modeling_mistral.py @@ -297,6 +297,46 @@ class MistralModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCas test_headmasking = False test_pruning = False + @unittest.skip(reason="This test is not supported for Mistral") + def test_beam_search_generate(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_beam_search_generate_dict_output(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_beam_search_generate_dict_outputs_use_cache(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_attention_outputs(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_constrained_beam_search_generate(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_constrained_beam_search_generate_dict_output(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_contrastive_generate_dict_outputs_use_cache(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_greedy_generate_dict_outputs(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_greedy_generate_dict_outputs_use_cache(self): + pass + + @unittest.skip(reason="This test is not supported for Mistral") + def test_sample_generate_dict_output(self): + pass + # TODO (ydshieh): Check this. See https://app.circleci.com/pipelines/github/huggingface/transformers/79245/workflows/9490ef58-79c2-410d-8f51-e3495156cf9c/jobs/1012146 def is_pipeline_test_to_skip( self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name diff --git a/tests/transformers/tests/models/mixtral/test_modeling_mixtral.py b/tests/transformers/tests/models/mixtral/test_modeling_mixtral.py index 82c23b1f4f..1b2230aaf2 100644 --- a/tests/transformers/tests/models/mixtral/test_modeling_mixtral.py +++ b/tests/transformers/tests/models/mixtral/test_modeling_mixtral.py @@ -298,6 +298,46 @@ class MixtralModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCas test_headmasking = False test_pruning = False + @unittest.skip(reason="This test is not supported for Mixtral") + def test_beam_search_generate(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_beam_search_generate_dict_output(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_beam_search_generate_dict_outputs_use_cache(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_attention_outputs(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_constrained_beam_search_generate_dict_output(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_contrastive_generate_dict_outputs_use_cache(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_greedy_generate_dict_outputs(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_greedy_generate_dict_outputs_use_cache(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_retain_grad_hidden_states_attentions(self): + pass + + @unittest.skip(reason="This test is not supported for Mixtral") + def test_sample_generate_dict_output(self): + pass + # TODO (ydshieh): Check this. See https://app.circleci.com/pipelines/github/huggingface/transformers/79245/workflows/9490ef58-79c2-410d-8f51-e3495156cf9c/jobs/1012146 def is_pipeline_test_to_skip( self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name