SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

better naming for if statement

Browse Source
This commit is contained in:
patrickvonplaten
2019-12-23 23:39:16 +01:00
parent 7e0c5c731a
commit d039c679d2
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/modeling_utils.py
View File

@@ -542,7 +542,11 @@ class PreTrainedModel(nn.Module):
def _do_output_past(self, outputs): 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 # TODO: might be better to write a self.do_output_past method for each individual class as is done for
# prepare_inputs_for_generation # prepare_inputs_for_generation
if hasattr(self.config, 'output_past') and self.config.output_past and len(outputs) > 1 and not hasattr(self, 'mem_len'): 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')
if has_output_past and has_multiple_outputs and not has_mem_len:
return True return True
# TODO: Add cases for (xlnet, transfo_xl) using mem_len # TODO: Add cases for (xlnet, transfo_xl) using mem_len
return False return False