From f02805da6fed33516fd77358068e459475c752ee Mon Sep 17 00:00:00 2001 From: thomwolf Date: Mon, 4 Nov 2019 15:42:23 +0100 Subject: [PATCH] fix tests --- transformers/tests/modeling_common_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/transformers/tests/modeling_common_test.py b/transformers/tests/modeling_common_test.py index 5fb029f454..159d9d85bb 100644 --- a/transformers/tests/modeling_common_test.py +++ b/transformers/tests/modeling_common_test.py @@ -492,11 +492,11 @@ class CommonTestCases: return equal for model_class in self.all_model_classes: - if not hasattr(model_class, 'tie_weights'): - continue - config.torchscript = True model_not_tied = model_class(config) + if model_not_tied.get_output_embeddings() is None: + continue + params_not_tied = list(model_not_tied.parameters()) config_tied = copy.deepcopy(config)