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 ) - - -