kv-cache : allow context shift for recurrent models

This commit is contained in:
Francis Couture-Harpin
2025-05-02 19:29:23 -04:00
parent d55b0d0621
commit e94f3932f2

View File

@ -1938,7 +1938,8 @@ llama_pos llama_kv_cache_recurrent::get_pos_max() const {
}
bool llama_kv_cache_recurrent::get_can_shift() const {
return false;
// shifting is trivial, the recurrent states don't care about the absolute position
return true;
}
uint32_t llama_kv_cache_recurrent::cell_max() const {