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

* add base model classes to  bart subclassed models

* add doc
This commit is contained in:
Suraj Patil
2020-12-21 19:56:46 +05:30
committed by GitHub
parent 08abdabda1
commit f4432b7e01
15 changed files with 134 additions and 17 deletions

View File

@@ -32,6 +32,7 @@ if is_torch_available():
AutoTokenizer,
BlenderbotConfig,
BlenderbotForConditionalGeneration,
BlenderbotModel,
BlenderbotSmallTokenizer,
BlenderbotTokenizer,
)
@@ -90,7 +91,7 @@ class BlenderbotModelTester:
class BlenderbotTesterMixin(ModelTesterMixin, unittest.TestCase):
if is_torch_available():
all_generative_model_classes = (BlenderbotForConditionalGeneration,)
all_model_classes = (BlenderbotForConditionalGeneration,)
all_model_classes = (BlenderbotForConditionalGeneration, BlenderbotModel)
else:
all_generative_model_classes = ()
all_model_classes = ()