riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[Tests] Improve flax test_attention_outputs (#21486)

Browse Source 
improving flax tests
This commit is contained in:
Shubhamai 2023-02-10 22:01:49 +05:30 committed by GitHub
parent c88b11c591
commit 9e40bba6ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_modeling_flax_common.py
View File

@ -650,6 +650,9 @@ class FlaxModelTesterMixin:
check_hidden_states_output(inputs_dict, config, model_class)
def test_attention_outputs(self):
if not self.has_attentions:
self.skipTest(reason="Model does not output attentions")
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
config.return_dict = True