Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down