Add Document QA pipeline metadata (#19028)

This commit is contained in:
Sylvain Gugger
2022-09-14 09:25:15 -04:00
committed by GitHub
parent 77b18783c2
commit 0b567aa430

View File

@@ -80,6 +80,11 @@ PIPELINE_TAGS_AND_AUTO_MODELS = [
"AutoModelForAudioFrameClassification", "AutoModelForAudioFrameClassification",
), ),
("audio-xvector", "MODEL_FOR_AUDIO_XVECTOR_MAPPING_NAMES", "AutoModelForAudioXVector"), ("audio-xvector", "MODEL_FOR_AUDIO_XVECTOR_MAPPING_NAMES", "AutoModelForAudioXVector"),
(
"document-question-answering",
"MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING_NAMES",
"AutoModelForDocumentQuestionAnswering",
),
] ]