kv-cache : fix can_shift() check to take into account M-RoPE (#19928)
This commit is contained in:
parent
b68d75165a
commit
99bd67c9b2
|
|
@ -978,6 +978,9 @@ bool llama_kv_cache::get_can_shift() const {
|
||||||
if (model.arch == LLM_ARCH_STEP35) {
|
if (model.arch == LLM_ARCH_STEP35) {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
if (hparams.n_pos_per_embd() > 1) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue