[bart] add config.extra_pos_embeddings to facilitate reuse (#5190)

2020-06-23 11:35:42 -04:00
parent b28b537131
commit 58918c76f4
5 changed files with 35 additions and 32 deletions
--- a/tests/test_modeling_roberta.py
+++ b/tests/test_modeling_roberta.py
@@ -34,9 +34,8 @@ if is_torch_available():
        RobertaForSequenceClassification,
        RobertaForTokenClassification,
    )
-    from transformers.modeling_roberta import RobertaEmbeddings
+    from transformers.modeling_roberta import RobertaEmbeddings, create_position_ids_from_input_ids
    from transformers.modeling_roberta import ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST
-    from transformers.modeling_utils import create_position_ids_from_input_ids


 class RobertaModelTester: