Default models for pipeline tasks
What are the default models used for the various pipeline tasks?  I assume the “SummarizationPipeline” uses Bart-large-cnn or some variant of T5, but what about the other tasks?|ConversationalPipeline|?||FeatureExtractionPipeline|?||FillMaskPipeline|?||QuestionAnsweringPipeline|?||SummarizationPipeline|BART or T5 (?)||TextClassificationPipeline|?||TextGenerationPipeline|?||TokenClassificationPipeline|?||TranslationPipeline|?||ZeroShotClassificationPipeline|?||Text2TextGenerationPipeline|?|
2020-12-04T18:12:57Z
You can find all the defaults here:github.comhuggingface/transformers/blob/71688a8889c4df7dd6d90a65d895ccf4e33a1a56/src/transformers/pipelines.py#L2716-L2804SUPPORTED_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
2023-12-27T14:31:38Z
Thanks.@BramVanroyAny idea on what basis these defaults were chosen for each task?