Update tiny model info. and pipeline testing (#25213)

* update tiny_model_summary.json * update * update * update --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-07-31 19:35:33 +02:00
parent e0c50b274a
commit 1b4f6199c6
8 changed files with 358 additions and 10 deletions
--- a/tests/models/falcon/test_modeling_falcon.py
+++ b/tests/models/falcon/test_modeling_falcon.py
@@ -278,9 +278,9 @@ class FalconModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMix
    pipeline_model_mapping = (
        {
            "feature-extraction": FalconModel,
+            "question-answering": FalconForQuestionAnswering,
            "text-classification": FalconForSequenceClassification,
            "text-generation": FalconForCausalLM,
-            "question-answering": FalconForQuestionAnswering,
            "token-classification": FalconForTokenClassification,
            "zero-shot": FalconForSequenceClassification,
        }