Add missing import (#8444)

* Add missing import * Fix dummy objects

Add missing import (#8444)
* Add missing import * Fix dummy objects
cace39af · Julien Plu · GitHub · e21340da · cace39af · cace39af
Unverified Commit cace39af authored Nov 10, 2020 by Julien Plu Committed by GitHub Nov 10, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

src/transformers/__init__.py src/transformers/__init__.py +1 -0

src/transformers/utils/dummy_tf_objects.py src/transformers/utils/dummy_tf_objects.py +3 -0

No files found.
--- a/src/transformers/__init__.py
+++ b/src/transformers/__init__.py
@@ -651,6 +651,7 @@ if is_tf_available():
        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,

--- a/src/transformers/utils/dummy_tf_objects.py
+++ b/src/transformers/utils/dummy_tf_objects.py
@@ -101,6 +101,9 @@ TF_MODEL_FOR_MASKED_LM_MAPPING = None
 TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING = None


+TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING = None
+
+
 TF_MODEL_FOR_PRETRAINING_MAPPING = None