skip two tests (#27013)

* skip two tests

* skip torch as well

* fixup
This commit is contained in:
Arthur
2023-10-23 12:52:05 +02:00
committed by GitHub
parent 45425660d0
commit ef978d0a7b
2 changed files with 12 additions and 1 deletions

View File

@@ -244,7 +244,9 @@ class BlenderbotSmallModelTest(ModelTesterMixin, GenerationTesterMixin, Pipeline
):
if pipeline_test_casse_name == "TextGenerationPipelineTests":
return True
# TODO @Rocketnight1 to fix
if pipeline_test_casse_name == "ConversationalPipelineTests":
return True
return False
def setUp(self):

View File

@@ -209,6 +209,15 @@ class TFBlenderbotSmallModelTest(TFModelTesterMixin, PipelineTesterMixin, unitte
config_and_inputs = self.model_tester.prepare_config_and_inputs_for_common()
self.model_tester.check_decoder_model_past_large_inputs(*config_and_inputs)
# TODO: Fix the failed tests when this model gets more usage
def is_pipeline_test_to_skip(
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name
):
# TODO @Rocketnight1 to fix
if pipeline_test_casse_name == "ConversationalPipelineTests":
return True
return False
@require_tokenizers
@require_tf