From ef978d0a7bb6455eff5c126cd6e4f10de0158004 Mon Sep 17 00:00:00 2001
From: Arthur <48595927+ArthurZucker@users.noreply.github.com>
Date: Mon, 23 Oct 2023 12:52:05 +0200
Subject: [PATCH] skip two tests (#27013)

* skip two tests

* skip torch as well

* fixup
---
 .../blenderbot_small/test_modeling_blenderbot_small.py   | 4 +++-
 .../test_modeling_tf_blenderbot_small.py                 | 9 +++++++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/tests/models/blenderbot_small/test_modeling_blenderbot_small.py b/tests/models/blenderbot_small/test_modeling_blenderbot_small.py
index 249a8a799a..2397b6fee9 100644
--- a/tests/models/blenderbot_small/test_modeling_blenderbot_small.py
+++ b/tests/models/blenderbot_small/test_modeling_blenderbot_small.py
@@ -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):
diff --git a/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py b/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py
index 021f171789..6fde705082 100644
--- a/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py
+++ b/tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py
@@ -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