From 8b3cf8cacdf7f81546f7fc612262a1a55ca7a462 Mon Sep 17 00:00:00 2001
From: Joshua Lochner <admin@xenova.com>
Date: Tue, 19 Sep 2023 01:29:17 +0200
Subject: [PATCH] Update list of supported models

---
 scripts/supported_models.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/scripts/supported_models.py b/scripts/supported_models.py
index 0b4312d34..9799ca8de 100644
--- 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',