Update tiny models and pipeline tests (#23446)

* fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-05-18 17:29:04 +02:00
parent 2406dbdcfa
commit ffad4f1373
9 changed files with 187 additions and 12 deletions
--- a/tests/models/biogpt/test_modeling_biogpt.py
+++ b/tests/models/biogpt/test_modeling_biogpt.py
@@ -289,9 +289,10 @@ class BioGptModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMix
    pipeline_model_mapping = (
        {
            "feature-extraction": BioGptModel,
+            "text-classification": BioGptForSequenceClassification,
            "text-generation": BioGptForCausalLM,
            "token-classification": BioGptForTokenClassification,
-            "text-classification": BioGptForSequenceClassification,
+            "zero-shot": BioGptForSequenceClassification,
        }
        if is_torch_available()
        else {}