kvcache: Set context for shift offsets

This commit is contained in:
Jesse Gross 2025-03-07 18:16:52 -08:00 committed by Jesse Gross
parent 25f9b152f9
commit f52b2615ef

View File

@ -545,7 +545,7 @@ func (c *Causal) shift(seq int, beginIndex, offset int32) error {
}
}
kShift, err := ctx.FromIntSlice(offsets, len(offsets))
kShift, err := ctx.Input().FromIntSlice(offsets, len(offsets))
if err != nil {
return err
}