Definitive HeisenDistilBug fix

cc @julien-c @@thomwolf
2020-01-27 12:09:58 -05:00
parent f09f42d4d3
commit 875c4ae48f
2 changed files with 5 additions and 1 deletions
--- a/tests/test_modeling_tf_distilbert.py
+++ b/tests/test_modeling_tf_distilbert.py
@@ -219,5 +219,5 @@ class TFDistilBertModelTest(TFModelTesterMixin, unittest.TestCase):
    # @slow
    # def test_model_from_pretrained(self):
    #     for model_name in list(DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
-    #         model = DistilBertModel.from_pretrained(model_name, cache_dir=CACHE_DIR)
+    #         model = DistilBertModesss.from_pretrained(model_name, cache_dir=CACHE_DIR)
    #         self.assertIsNotNone(model)