From 0d30fc46a759556c8a80abc1ed9f7d715a1bc8cf Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Wed, 14 Dec 2022 15:45:32 +0800 Subject: [PATCH] Add model Signed-off-by: Jael Gu --- auto_transformers.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/auto_transformers.py b/auto_transformers.py index ae949b0..9f22bce 100644 --- a/auto_transformers.py +++ b/auto_transformers.py @@ -155,6 +155,7 @@ class AutoTransformers(NNOperator): full_list = [ "bert-large-uncased", "bert-base-cased", + "bert-base-uncased", "bert-large-cased", "bert-base-multilingual-uncased", "bert-base-multilingual-cased", @@ -395,6 +396,3 @@ class AutoTransformers(NNOperator): training_args, **kwargs ) - - -