only do output_past=True for language generation in bart

2020-03-05 16:34:47 +01:00
parent 7cba11fb9b
commit aceb3fbaf4
2 changed files with 3 additions and 2 deletions
--- a/tests/test_modeling_bart.py
+++ b/tests/test_modeling_bart.py
@@ -64,7 +64,6 @@ class ModelTester:
        self.eos_token_id = 2
        self.pad_token_id = 1
        self.bos_token_id = 0
-        self.output_past = True
        torch.manual_seed(0)

    def prepare_config_and_inputs_for_common(self):
@@ -86,7 +85,6 @@ class ModelTester:
            eos_token_ids=self.eos_token_id,
            bos_token_id=self.bos_token_id,
            pad_token_id=self.pad_token_id,
-            output_past=self.output_past,
        )
        inputs_dict = prepare_bart_inputs_dict(config, input_ids)
        return config, inputs_dict