Skip to content

Commit 14a6b6c

Browse files
authored
Merge pull request #72 from FluffyAIcode/AgentMemory/v04-pr-k1c-restored-attention-8e7f
PR-K1.C: per-attention-layer K/V preparation primitive (stacked on #71)
2 parents 529abcf + 651305b commit 14a6b6c

3 files changed

Lines changed: 913 additions & 0 deletions

File tree

inference_engine/v04/__init__.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,11 @@
3737
compute_evicted_positions,
3838
merge_kv_at_evicted_positions,
3939
)
40+
from inference_engine.v04.restored_attention import (
41+
apply_rope_to_k_at_positions,
42+
prepare_restored_attention_kv,
43+
slice_position_embeddings,
44+
)
4045

4146
__all__ = [
4247
# K1.A — capture
@@ -46,4 +51,8 @@
4651
# K1.B — merge
4752
"compute_evicted_positions",
4853
"merge_kv_at_evicted_positions",
54+
# K1.C — restored attention K/V preparation
55+
"apply_rope_to_k_at_positions",
56+
"prepare_restored_attention_kv",
57+
"slice_position_embeddings",
4958
]

0 commit comments

Comments
 (0)