Browse Source
Add model
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
1 additions and
3 deletions
-
auto_transformers.py
|
@ -155,6 +155,7 @@ class AutoTransformers(NNOperator): |
|
|
full_list = [ |
|
|
full_list = [ |
|
|
"bert-large-uncased", |
|
|
"bert-large-uncased", |
|
|
"bert-base-cased", |
|
|
"bert-base-cased", |
|
|
|
|
|
"bert-base-uncased", |
|
|
"bert-large-cased", |
|
|
"bert-large-cased", |
|
|
"bert-base-multilingual-uncased", |
|
|
"bert-base-multilingual-uncased", |
|
|
"bert-base-multilingual-cased", |
|
|
"bert-base-multilingual-cased", |
|
@ -395,6 +396,3 @@ class AutoTransformers(NNOperator): |
|
|
training_args, |
|
|
training_args, |
|
|
**kwargs |
|
|
**kwargs |
|
|
) |
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|