Added integration to masked_language_modeling training command (#708)

anmarques · web-flow · commit 32e6d84fe58e · 2022-04-14T12:09:11.000-04:00
diff --git a/setup.py b/setup.py
@@ -132,6 +132,7 @@ def _setup_entry_points() -> Dict:
 
     # transformers integration
     for task in [
+        "masked_language_modeling",
         "question_answering",
         "text_classification",
         "token_classification",
