@@ -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):
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user