Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "73fcfb286184114207da5b6400d78a7f7e0fa79e"
Commit cbcb83f2 authored by sshleifer's avatar sshleifer Committed by Lysandre Debut
Browse files

minor cleanup of test_attention_outputs

parent 3bf54172
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 17 deletions
tests/test_modeling_common.py
View file @ cbcb83f2
......@@ -117,23 +117,11 @@ class ModelTesterMixin:
def test_attention_outputs(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
decoder_seq_length = (
self.model_tester.decoder_seq_length
if hasattr(self.model_tester, "decoder_seq_length")
else self.model_tester.seq_length
)
encoder_seq_length = (
self.model_tester.encoder_seq_length
if hasattr(self.model_tester, "encoder_seq_length")
else self.model_tester.seq_length
)
decoder_key_length = (
self.model_tester.key_length if hasattr(self.model_tester, "key_length") else decoder_seq_length
)
encoder_key_length = (
self.model_tester.key_length if hasattr(self.model_tester, "key_length") else encoder_seq_length
)
seq_len = self.model_tester.seq_length
decoder_seq_length = getattr(self.model_tester, 'decoder_seq_length', seq_len)
encoder_seq_length = getattr(self.model_tester, 'encoder_seq_length', seq_len)
decoder_key_length = getattr(self.model_tester, "key_length", decoder_seq_length)
encoder_key_length = getattr(self.model_tester, "key_length", encoder_seq_length)
for model_class in self.all_model_classes:
config.output_attentions = True
......
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