speculative : fix compile warning
Some checks are pending
flake8 Lint / Lint (push) Waiting to run

This commit is contained in:
Georgi Gerganov 2024-11-24 12:53:48 +02:00
parent d9fb3b2e01
commit c8880e786c
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -170,7 +170,7 @@ llama_tokens common_speculative_gen_draft(
for (int i = reuse_i + reuse_n + 1; i < (int) prompt.size(); ++i) {
result.push_back(prompt[i]);
if (result.size() >= params.n_draft) {
if (params.n_draft <= (int) result.size()) {
break;
}
}