-
Notifications
You must be signed in to change notification settings - Fork 447
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
The get_ppl
missed the last token of each iteration during multi-iter prefill
#2499
Conversation
@grimoire can we remove the |
Accept |
get_ppl
missed the last token of each iteration during multi-iter prefill
LGTM the unit-test error is because the latest transformers changed their api and our logic for getting stop words invoked that function. https://github.com/huggingface/transformers/blob/v4.44.2/src/transformers/tokenization_utils_base.py#L3526 |
logits
is 2G, preventing OOM whenbatch_size * seq_len
is too large