From b889d3f6c4ef3325698e0a40667fbfa6200b40c2 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Fri, 2 Jul 2021 10:35:10 +0200 Subject: [PATCH] Fix TAPAS test uncovered by #12446 (#12480) --- tests/test_modeling_tapas.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_tapas.py b/tests/test_modeling_tapas.py index 40bdba0e70..32265c25f6 100644 --- a/tests/test_modeling_tapas.py +++ b/tests/test_modeling_tapas.py @@ -106,7 +106,7 @@ class TapasModelTester: average_logits_per_cell=True, select_one_column=True, allow_empty_column_selection=False, - init_cell_selection_weights_to_zero=False, + init_cell_selection_weights_to_zero=True, reset_position_index_per_cell=True, disable_per_token_loss=False, scope=None,