Fix TAPAS test uncovered by #12446 (#12480)

b889d3f6 · Lysandre Debut · GitHub · b4ecc6be · b889d3f6
Unverified Commit b889d3f6 authored Jul 02, 2021 by Lysandre Debut Committed by GitHub Jul 02, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_tapas.py tests/test_modeling_tapas.py +1 -1

No files found.
--- 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,