Default models for pipeline tasks

BramVanroy · December 4, 2020, 6:12pm

You can find all the defaults here:

huggingface/transformers/blob/71688a8889c4df7dd6d90a65d895ccf4e33a1a56/src/transformers/pipelines.py#L2716-L2804


SUPPORTED_TASKS = {
    "feature-extraction": {
        "impl": FeatureExtractionPipeline,
        "tf": TFAutoModel if is_tf_available() else None,
        "pt": AutoModel if is_torch_available() else None,
        "default": {"model": {"pt": "distilbert-base-cased", "tf": "distilbert-base-cased"}},
    },
    "sentiment-analysis": {
        "impl": TextClassificationPipeline,
        "tf": TFAutoModelForSequenceClassification if is_tf_available() else None,
        "pt": AutoModelForSequenceClassification if is_torch_available() else None,
        "default": {
            "model": {
                "pt": "distilbert-base-uncased-finetuned-sst-2-english",
                "tf": "distilbert-base-uncased-finetuned-sst-2-english",
            },
        },
    },
    "ner": {
        "impl": TokenClassificationPipeline,

This file has been truncated. show original

Topic		Replies	Views
Looking for tool class to do predictions 🤗Transformers	3	552	October 9, 2020
[RFC] Transformers Pipeline v2 🤗Transformers	4	1851	October 14, 2020
Using Huggingface for computer vision (Tensorflow)? 🤗Transformers	3	417	June 2, 2025
New pipeline for zero-shot text classification 🤗Transformers	107	71764	February 17, 2025
What's the pipeline task string for paraphrase sentence detection Beginners	0	189	December 31, 2023

Default models for pipeline tasks

Related topics