Unverified Commit 33929448 authored by Nils Reimers's avatar Nils Reimers Committed by GitHub
Browse files

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

parent a6d62aab
......@@ -1835,7 +1835,7 @@ class GenerationMixin:
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
# 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