Skip to content

Commit ae95a7c

Browse files
committed
fix: remove non-existent self_kq_mask_swa from attn_kv
1 parent 46394ed commit ae95a7c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/llama-context.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1699,7 +1699,7 @@ llm_graph_result * llama_context::process_ubatch(const llama_ubatch & ubatch, ll
16991699
auto * base = inp.get();
17001700
if (auto * akv = dynamic_cast<llm_graph_input_attn_kv *>(base)) {
17011701
force(akv->self_k_idxs); force(akv->self_v_idxs);
1702-
force(akv->self_kq_mask); force(akv->self_kq_mask_swa);
1702+
force(akv->self_kq_mask);
17031703
} else if (auto * ak = dynamic_cast<llm_graph_input_attn_k *>(base)) {
17041704
force(ak->self_k_idxs); force(ak->self_kq_mask);
17051705
} else if (auto * dsa = dynamic_cast<llm_graph_input_attn_k_dsa *>(base)) {

0 commit comments

Comments
 (0)