From 0de885a6ac2d0f219dde5ca62fe932806122a60b Mon Sep 17 00:00:00 2001 From: Joshua Lochner Date: Mon, 11 Sep 2023 01:25:48 +0200 Subject: [PATCH] Add blenderbot to supported models --- scripts/supported_models.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/scripts/supported_models.py b/scripts/supported_models.py index 0b4312d..c998ef9 100644 --- a/scripts/supported_models.py +++ b/scripts/supported_models.py @@ -97,16 +97,16 @@ SUPPORTED_MODELS = { 'bert-base-chinese', 'emilyalsentzer/Bio_ClinicalBERT', ], - # 'blenderbot': [ - # # Text2text generation (TODO add conversational) - # 'facebook/blenderbot-400M-distill', - # 'facebook/blenderbot-1B-distill', - # ], - # 'blenderbot-small': [ - # # Text2text generation (TODO add conversational) - # 'facebook/blenderbot-90M', # DEPRECATED - # 'facebook/blenderbot_small-90M', - # ], + 'blenderbot': [ + # Text2text generation (TODO add conversational) + 'facebook/blenderbot-400M-distill', + # 'facebook/blenderbot-1B-distill', + ], + 'blenderbot-small': [ + # Text2text generation (TODO add conversational) + # 'facebook/blenderbot-90M', # DEPRECATED + 'facebook/blenderbot_small-90M', + ], 'bloom': [ # Text generation 'bigscience/bloom-560m',