Skip to content

GitLab

Commit eeaa402c authored by patrickvonplaten's avatar patrickvonplaten
Browse files

rename comments

parent 7bb42712
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/modeling_utils.py
View file @ eeaa402c
...@@ -731,7 +731,7 @@ class PreTrainedModel(nn.Module): ...@@ -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): ...@@ -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]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment