Skip to content

GitLab

Commit d997ac78 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

fix typo

parent 7351a8db
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_utils.py
View file @ d997ac78
...@@ -1281,7 +1281,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin): ...@@ -1281,7 +1281,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin):
if self.config.is_encoder_decoder: if self.config.is_encoder_decoder:
# do not return first <EOS> token # do not return first <EOS> token
return decoded[:, 1:] return decoded[:, 1:]
# return decoded return decoded
# force one of token_ids to be generated by setting prob of all other tokens to 0. # force one of token_ids to be generated by setting prob of all other tokens to 0.
def _force_token_ids_generation(self, scores, token_ids): def _force_token_ids_generation(self, scores, token_ids):
......
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