Skip to content

GitLab

Commit 897a24c8 authored by Lysandre's avatar Lysandre
Browse files

Fix head_mask for model templates

parent 10e5f282
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py
View file @ 897a24c8
...@@ -252,6 +252,8 @@ class TF{{cookiecutter.camelcase_modelname}}ModelTest(TFModelTesterMixin, unitte ...@@ -252,6 +252,8 @@ class TF{{cookiecutter.camelcase_modelname}}ModelTest(TFModelTesterMixin, unitte
else () else ()
) )
test_head_masking = False
def setUp(self): def setUp(self):
self.model_tester = TF{{cookiecutter.camelcase_modelname}}ModelTester(self) self.model_tester = TF{{cookiecutter.camelcase_modelname}}ModelTester(self)
self.config_tester = ConfigTester(self, config_class={{cookiecutter.camelcase_modelname}}Config, hidden_size=37) self.config_tester = ConfigTester(self, config_class={{cookiecutter.camelcase_modelname}}Config, hidden_size=37)
...@@ -475,6 +477,7 @@ class TF{{cookiecutter.camelcase_modelname}}ModelTest(TFModelTesterMixin, unitte ...@@ -475,6 +477,7 @@ class TF{{cookiecutter.camelcase_modelname}}ModelTest(TFModelTesterMixin, unitte
all_generative_model_classes = (TF{{cookiecutter.camelcase_modelname}}ForConditionalGeneration,) if is_tf_available() else () all_generative_model_classes = (TF{{cookiecutter.camelcase_modelname}}ForConditionalGeneration,) if is_tf_available() else ()
is_encoder_decoder = True is_encoder_decoder = True
test_pruning = False test_pruning = False
test_head_masking = False
def setUp(self): def setUp(self):
self.model_tester = TF{{cookiecutter.camelcase_modelname}}ModelTester(self) self.model_tester = TF{{cookiecutter.camelcase_modelname}}ModelTester(self)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment