kv-cache : fix can_shift() check to take into account M-RoPE (#19928)

This commit is contained in:
Georgi Gerganov 2026-02-26 18:08:54 +02:00 committed by GitHub
parent b68d75165a
commit 99bd67c9b2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions

View File

@ -978,6 +978,9 @@ bool llama_kv_cache::get_can_shift() const {
if (model.arch == LLM_ARCH_STEP35) {
return false;
}
if (hparams.n_pos_per_embd() > 1) {
return false;
}
return true;
}