ver4a/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

hparams : add SWA rope parameters (#12374)

Browse source 
ggml-ci
This commit is contained in:
Georgi Gerganov 2025-03-14 09:03:24 +02:00 • committed by GitHub
parent 84d5475541
commit 081bee8c64
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 26 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/llama-graph.cpp
View file

@ -1403,9 +1403,9 @@ ggml_tensor * llm_graph_context::build_attn(
ggml_build_forward_expand(gf, ggml_cpy(ctx0, v_cur, v_cache_view));
}
const bool is_sliding = hparams.is_sliding(il);
const bool is_swa = hparams.is_swa(il);
const auto & kq_mask = is_sliding ? inp->get_kq_mask_swa() : inp->get_kq_mask();
const auto & kq_mask = is_swa ? inp->get_kq_mask_swa() : inp->get_kq_mask();
const auto n_kv = kv_self->n;