pub struct VitArgs<'a> {}Expand description
A vision tower’s encoder stack for vit_encode. cos/sin are
[n, head_dim] per lane; rope_halves pairs lane j with j ± hd/2
(Qwen-VL), else with its neighbour (OneVision’s interleaved form).
key identifies the tower for the device weight cache.
Fields§
§key: u64§layers: Vec<VitLayer<'a>>§n: usize§d: usize§heads: usize§head_dim: usize§inter: usize§eps: f64§gelu_tanh: bool§rope_halves: bool§cos: &'a [f32]§sin: &'a [f32]§prof: boolAuto Trait Implementations§
impl<'a> Freeze for VitArgs<'a>
impl<'a> RefUnwindSafe for VitArgs<'a>
impl<'a> Send for VitArgs<'a>
impl<'a> Sync for VitArgs<'a>
impl<'a> Unpin for VitArgs<'a>
impl<'a> UnsafeUnpin for VitArgs<'a>
impl<'a> UnwindSafe for VitArgs<'a>
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