Reduce the time spent for the TF slow tests (#10152)

* rework savedmodel slow test * Improve savedmodel tests * Remove useless content
2021-02-18 15:52:57 +01:00
parent 14ed3b978e
commit 2acae50a0c
7 changed files with 91 additions and 166 deletions
--- a/tests/test_modeling_tf_t5.py
+++ b/tests/test_modeling_tf_t5.py
@@ -237,7 +237,6 @@ class TFT5ModelTester:
            "input_ids": input_ids,
            "decoder_input_ids": input_ids,
            "decoder_attention_mask": input_mask,
-            "use_cache": False,
        }
        return config, inputs_dict