add base model classes to bart subclassed models (#9230)

* add base model classes to bart subclassed models * add doc
2020-12-21 19:56:46 +05:30
parent 08abdabda1
commit f4432b7e01
15 changed files with 134 additions and 17 deletions
--- a/tests/test_modeling_mbart.py
+++ b/tests/test_modeling_mbart.py
@@ -30,6 +30,7 @@ if is_torch_available():
        BatchEncoding,
        MBartConfig,
        MBartForConditionalGeneration,
+        MBartModel,
    )


@@ -59,7 +60,7 @@ class ModelTester:

@require_torch
 class SelectiveCommonTest(unittest.TestCase):
-    all_model_classes = (MBartForConditionalGeneration,) if is_torch_available() else ()
+    all_model_classes = (MBartForConditionalGeneration, MBartModel) if is_torch_available() else ()

    test_save_load__keys_to_ignore_on_save = ModelTesterMixin.test_save_load__keys_to_ignore_on_save