Add TFCamembertForCausalLM and ONNX integration test (#16073)

* Make Camembert great again!

* Add Camembert to TensorFlow ONNX tests
This commit is contained in:
lewtun
2022-03-14 08:40:42 +01:00
committed by GitHub
parent 20ab1582cf
commit 6e1e88fd38
7 changed files with 30 additions and 0 deletions

View File

@@ -200,6 +200,7 @@ PYTORCH_EXPORT_SEQ2SEQ_WITH_PAST_MODELS = {
TENSORFLOW_EXPORT_DEFAULT_MODELS = {
("albert", "hf-internal-testing/tiny-albert"),
("bert", "bert-base-cased"),
("camembert", "camembert-base"),
("distilbert", "distilbert-base-cased"),
("roberta", "roberta-base"),
}