Skip 2 failing persimmon pipeline tests for now (#26485)

skip Co-authored-by: ydshieh <[email protected]>
huggingface · Sep 29, 2023 · 9b23d0d · 9b23d0d
1 parent 14170b7
commit 9b23d0d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/models/persimmon/test_modeling_persimmon.py b/tests/models/persimmon/test_modeling_persimmon.py
@@ -276,8 +276,9 @@ class PersimmonModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTester
         {
             "feature-extraction": PersimmonModel,
             "text-classification": PersimmonForSequenceClassification,
-            "text-generation": PersimmonForCausalLM,
-            "zero-shot": PersimmonForSequenceClassification,
+            # TODO (ydshieh): check why these two fail. Fix them or skip them in a better way.
+            # "text-generation": PersimmonForCausalLM,
+            # "zero-shot": PersimmonForSequenceClassification,
         }
         if is_torch_available()
         else {}