Add Document QA pipeline metadata (#19028)

0b567aa4 · Sylvain Gugger · GitHub · 77b18783 · 0b567aa4
Unverified Commit 0b567aa4 authored Sep 14, 2022 by Sylvain Gugger Committed by GitHub Sep 14, 2022
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

utils/update_metadata.py utils/update_metadata.py +5 -0

No files found.
--- a/utils/update_metadata.py
+++ b/utils/update_metadata.py
@@ -80,6 +80,11 @@ PIPELINE_TAGS_AND_AUTO_MODELS = [
        "AutoModelForAudioFrameClassification",
    ),
    ("audio-xvector", "MODEL_FOR_AUDIO_XVECTOR_MAPPING_NAMES", "AutoModelForAudioXVector"),
+    (
+        "document-question-answering",
+        "MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING_NAMES",
+        "AutoModelForDocumentQuestionAnswering",
+    ),
 ]