SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
thomwolf
2019-11-04 15:42:23 +01:00
parent 1724cee8c4
commit f02805da6f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
transformers/tests/modeling_common_test.py
View File

@@ -492,11 +492,11 @@ class CommonTestCases:
return equal return equal
for model_class in self.all_model_classes: for model_class in self.all_model_classes:
if not hasattr(model_class, 'tie_weights'):
continue
config.torchscript = True config.torchscript = True
model_not_tied = model_class(config) model_not_tied = model_class(config)
if model_not_tied.get_output_embeddings() is None:
continue
params_not_tied = list(model_not_tied.parameters()) params_not_tied = list(model_not_tied.parameters())
config_tied = copy.deepcopy(config) config_tied = copy.deepcopy(config)