pub struct BoundedWeights {
pub wq: QTensor,
pub wk: QTensor,
pub wv: QTensor,
pub wo: QTensor,
pub sink_k: Vec<f32>,
pub sink_v: Vec<f32>,
pub sink: usize,
pub window: usize,
pub group_windows: Vec<usize>,
}Expand description
Weights of one bounded-anchor layer.
Fields§
§wq: QTensor§wk: QTensor§wv: QTensor§wo: QTensor§sink_k: Vec<f32>Trained NoPE sink keys [kvh][sink][hd] (weights, not positions).
sink_v: Vec<f32>Trained sink values [kvh][sink][hd].
sink: usize§window: usizeRing capacity: the widest group window.
group_windows: Vec<usize>swa_sink_v2: the exact window of every KV group ([kvh], each
≤ window); empty = every group attends to the whole window (v1).
Auto Trait Implementations§
impl Freeze for BoundedWeights
impl RefUnwindSafe for BoundedWeights
impl Send for BoundedWeights
impl Sync for BoundedWeights
impl Unpin for BoundedWeights
impl UnsafeUnpin for BoundedWeights
impl UnwindSafe for BoundedWeights
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more