ALBERT | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
AST | feature-extraction, audio-classification |
BERT | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
BLOOM | text-generation |
Beit | feature-extraction, image-classification |
CamemBERT | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
ConvBERT | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
ConvNext | feature-extraction, image-classification |
ConvNextV2 | feature-extraction, image-classification |
CvT | feature-extraction, image-classification |
DeBERTa (INF2 only) | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
DeBERTa-v2 (INF2 only) | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
Deit | feature-extraction, image-classification |
DistilBERT | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
DonutSwin | feature-extraction |
Dpt | feature-extraction |
ELECTRA | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
ESM | feature-extraction, fill-mask, text-classification, token-classification |
FlauBERT | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
GPT2 | text-generation |
Hubert | feature-extraction, automatic-speech-recognition, audio-classification |
Levit | feature-extraction, image-classification |
Llama, Llama 2, Llama 3 | text-generation |
Mistral | text-generation |
Mixtral | text-generation |
MobileBERT | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
MobileNetV2 | feature-extraction, image-classification, semantic-segmentation |
MobileViT | feature-extraction, image-classification, semantic-segmentation |
MPNet | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
OPT | text-generation |
Phi | feature-extraction, text-classification, token-classification |
RoBERTa | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
RoFormer | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
Swin | feature-extraction, image-classification |
T5 | text2text-generation |
UniSpeech | feature-extraction, automatic-speech-recognition, audio-classification |
UniSpeech-SAT | feature-extraction, automatic-speech-recognition, audio-classification, audio-frame-classification, audio-xvector |
ViT | feature-extraction, image-classification |
Wav2Vec2 | feature-extraction, automatic-speech-recognition, audio-classification, audio-frame-classification, audio-xvector |
WavLM | feature-extraction, automatic-speech-recognition, audio-classification, audio-frame-classification, audio-xvector |
XLM | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
XLM-RoBERTa | feature-extraction, fill-mask, multiple-choice, question-answering, text-classification, token-classification |
Yolos | feature-extraction, object-detection |