dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
utils/update_metadata.py
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",
),
] ]