Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "9853c5dd587b9fb5eb3e39f9282c3c9e56188eb0"
Unverified Commit 33929448 authored by Nils Reimers's avatar Nils Reimers Committed by GitHub
Browse files

Replace // operator with / operator + long() (#13013)

parent a6d62aab
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/generation_utils.py
View file @ 33929448
...@@ -1835,7 +1835,7 @@ class GenerationMixin: ...@@ -1835,7 +1835,7 @@ class GenerationMixin:
next_token_scores, 2 * num_beams, dim=1, largest=True, sorted=True next_token_scores, 2 * num_beams, dim=1, largest=True, sorted=True
) )
next_indices = next_tokens // vocab_size next_indices = (next_tokens / vocab_size).long()
next_tokens = next_tokens % vocab_size next_tokens = next_tokens % vocab_size
# stateless # stateless
......
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