make if statements cleaner for prepare_inputs_for_generation

This commit is contained in:
patrickvonplaten
2019-12-23 23:55:05 +01:00
parent d039c679d2
commit 365ccd0af2
3 changed files with 8 additions and 6 deletions

View File

@@ -540,8 +540,8 @@ class PreTrainedModel(nn.Module):
return {"input_ids": input_ids}
def _do_output_past(self, outputs):
# TODO: might be better to write a self.do_output_past method for each individual class as is done for
# prepare_inputs_for_generation
# TODO: might be better to write a self.do_output_past method for each
# individual class as is done for prepare_inputs_for_generation
has_output_past = hasattr(self.config, 'output_past') and self.config.output_past
has_multiple_outputs = len(outputs) > 1
has_mem_len = hasattr(self, 'mem_len')