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',