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

rename comments

Browse Source
This commit is contained in:
patrickvonplaten
2019-12-23 22:15:06 +01:00
parent 7bb4271291
commit eeaa402cd4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/modeling_utils.py
View File

@@ -731,7 +731,7 @@ class PreTrainedModel(nn.Module):
outputs = self(**model_inputs) outputs = self(**model_inputs)
next_token_logits = outputs[0][:, -1, :] next_token_logits = outputs[0][:, -1, :]
# if model has past, then set the past parameter to speed up decoding # if model has past, then set the past variable to speed up decoding
if self._has_past(outputs): if self._has_past(outputs):
past = outputs[1] past = outputs[1]
@@ -818,7 +818,7 @@ class PreTrainedModel(nn.Module):
outputs = self(**model_inputs) # (batch_size * num_beams, cur_len, vocab_size) outputs = self(**model_inputs) # (batch_size * num_beams, cur_len, vocab_size)
scores = outputs[0][:, -1, :] # (batch_size * num_beams, vocab_size) scores = outputs[0][:, -1, :] # (batch_size * num_beams, vocab_size)
# if model has past, then set the past parameter to speed up decoding # if model has past, then set the past variable to speed up decoding
if self._has_past(outputs): if self._has_past(outputs):
past = outputs[1] past = outputs[1]