Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "d90a36d192e2981a41122c30a765c63158dd0557"
Unverified Commit 5a6b3ccd authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Skip equivalence test for TransfoXL (#16224) 

* Skip test for TransfoXL

* Single list
parent abd503d9
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
tests/test_modeling_tf_common.py
View file @ 5a6b3ccd
...@@ -476,6 +476,7 @@ class TFModelTesterMixin: ...@@ -476,6 +476,7 @@ class TFModelTesterMixin:
"TFFunnelForPreTraining", "TFFunnelForPreTraining",
"TFElectraForPreTraining", "TFElectraForPreTraining",
"TFXLMWithLMHeadModel", "TFXLMWithLMHeadModel",
"TFTransfoXLLMHeadModel",
]: ]:
self.assertEqual(tf_loss is None, pt_loss is None) self.assertEqual(tf_loss is None, pt_loss is None)
...@@ -490,7 +491,8 @@ class TFModelTesterMixin: ...@@ -490,7 +491,8 @@ class TFModelTesterMixin:
"TFFunnelForPreTraining", "TFFunnelForPreTraining",
"TFElectraForPreTraining", "TFElectraForPreTraining",
"TFXLMWithLMHeadModel", "TFXLMWithLMHeadModel",
] + ["TFTransfoXLLMHeadModel"]: "TFTransfoXLLMHeadModel",
]:
self.assertEqual(tf_keys, pt_keys) self.assertEqual(tf_keys, pt_keys)
# Since we deliberately make some tests pass above (regarding the `loss`), let's still try to test # Since we deliberately make some tests pass above (regarding the `loss`), let's still try to test
......
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