Update list of supported pix2struct models

huggingface · Jan 16, 2024 · 5bddf7b · 5bddf7b
1 parent c42d8e7
commit 5bddf7b
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/scripts/supported_models.py b/scripts/supported_models.py
@@ -732,14 +732,12 @@
         # Visual Question Answering (VQA)
         'visual-question-answering': [
             'google/deplot',
-
-            # TODO:
-            # 'google/pix2struct-docvqa-base',
-            # 'google/pix2struct-widget-captioning-base',
-            # 'google/pix2struct-ai2d-base',
-            # 'google/pix2struct-chartqa-base',
-            # 'google/pix2struct-screen2words-base',
-            # 'google/pix2struct-infographics-vqa-base',
+            'google/pix2struct-docvqa-base',
+            'google/pix2struct-widget-captioning-base',
+            'google/pix2struct-ai2d-base',
+            'google/pix2struct-chartqa-base',
+            'google/pix2struct-screen2words-base',
+            'google/pix2struct-infographics-vqa-base',
         ],
     },
     'roberta': {