Add missing import (#8444)

* Add missing import * Fix dummy objects
huggingface · Nov 10, 2020 · cace39a · cace39a
1 parent e21340d
commit cace39a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -651,6 +651,7 @@
         TF_MODEL_FOR_CAUSAL_LM_MAPPING,
         TF_MODEL_FOR_MASKED_LM_MAPPING,
         TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING,
+        TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING,
         TF_MODEL_FOR_PRETRAINING_MAPPING,
         TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING,
         TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING,

diff --git a/src/transformers/utils/dummy_tf_objects.py b/src/transformers/utils/dummy_tf_objects.py
@@ -101,6 +101,9 @@ def from_pretrained(self, *args, **kwargs):
 TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING = None
 
 
+TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING = None
+
+
 TF_MODEL_FOR_PRETRAINING_MAPPING = None