it s the complete opposite...

hyenal · Apr 30, 2024 · 61a97df · 61a97df
1 parent aefab37
commit 61a97df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/models/vision_encoder_decoder/test_modeling_tf_vision_encoder_decoder.py b/tests/models/vision_encoder_decoder/test_modeling_tf_vision_encoder_decoder.py
@@ -556,8 +556,8 @@ def test_pt_tf_model_equivalence(self):
         config.output_hidden_states = True
         decoder_config.output_hidden_states = True
         # All models tested in this file have attentions in slow mode
-        config.output_attentions = not _run_slow_tests
-        decoder_config.output_attentions = not _run_slow_tests
+        config.output_attentions = _run_slow_tests
+        decoder_config.output_attentions = _run_slow_tests
 
         tf_inputs_dict = config_inputs_dict
         # `encoder_hidden_states` is not used in model call/forward