From 44c07c7b7012dfad338917d6a72c1f83727d4f6c Mon Sep 17 00:00:00 2001
From: Alexandre Marques <alexandre@neuralmagic.com>
Date: Thu, 14 Apr 2022 11:25:14 -0400
Subject: [PATCH] Added integration to masked_language_modeling training
 command

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index ae30d8ede6f..41acb259164 100644
--- 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",