Fix TF template (huggingface#10189)

* Fix template * Update Seq2Seq tests
fadi212 · Feb 15, 2021 · 5702188 · 5702188
1 parent 2a5c990
commit 5702188
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...plate-{{cookiecutter.modelname}}/test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py b/...plate-{{cookiecutter.modelname}}/test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py
@@ -253,6 +253,7 @@ class TF{{cookiecutter.camelcase_modelname}}ModelTest(TFModelTesterMixin, unitte
     )
 
     test_head_masking = False
+    test_onnx = False
 
     def setUp(self):
         self.model_tester = TF{{cookiecutter.camelcase_modelname}}ModelTester(self)
@@ -478,6 +479,7 @@ class TF{{cookiecutter.camelcase_modelname}}ModelTest(TFModelTesterMixin, unitte
     is_encoder_decoder = True
     test_pruning = False
     test_head_masking = False
+    test_onnx = False
 
     def setUp(self):
         self.model_tester = TF{{cookiecutter.camelcase_modelname}}ModelTester(self)