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?
Hi Bram. Many thanks for pointing to the repo. I already learned yesterday about sentiment-analysis being an “alias” to the text-analysis pipeline. I am interested in the same question as the original poster was. Finding out what the default models for the various tasks I can use with the pipeline library. If you use CTRL-SPACE in VSCode to get autocompletion for the pipeline item, you get about 30 different options you can use with the pipeline. Things like “audio-classification”, “depht-estimation”, “image-to-text”, visual question.-answering and many more. I added screenshots about all the options I get from this. I tried to find them in the transformers/pipeline.py file but I missed. Where could I find the corresponding default models used for these tasks? Not truly urgent but I am building a GUI for handling pipeline tasks and I would like to download the correct models upfront for offline use. Many thanks for any hints.
Yes, exactly. There is no pipelines.py file anymore, the relevant code has been moved to __init__.py. The pre-configs are all stored in the SUPPORTED_TASKS variable, as already explained here:
Many thanks for your patience, sir. I missed that the other “aliases” could be in the same file you told me about when I asked about “sentiment-analysis”. Should have come to my mind by myself to check this. Thanks.