Update list of supported models

huggingface · Sep 18, 2023 · 8b3cf8c · 8b3cf8c
1 parent db42a11
commit 8b3cf8c
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/scripts/supported_models.py b/scripts/supported_models.py
@@ -265,6 +265,27 @@
         'JackFram/llama-68m',
         'JackFram/llama-160m',
     ],
+    'longt5': {
+        # Text-to-text
+        'text2text-generation': [
+            'google/long-t5-local-base',
+            'google/long-t5-tglobal-base',
+            # 'google/long-t5-tglobal-xl', # too large
+            # 'google/long-t5-tglobal-large', # too large
+            # 'google/long-t5-local-large', # too large
+        ],
+
+        # Summarization
+        'summarization': [
+            'pszemraj/long-t5-tglobal-base-16384-book-summary',
+        ],
+
+        # Feature extraction
+        'feature-extraction': [
+            # NOTE: requires --task feature-extraction
+            'voidful/long-t5-encodec-tglobal-base',
+        ],
+    },
     'm2m_100': [
         # Translation
         'facebook/nllb-200-distilled-600M',