GlmVitMerger

metrale_model_layers::layers::glm_vit

Struct GlmVitMerger

pub struct GlmVitMerger {
    pub post_layernorm_w: DevicePtr,
    pub downsample_w: DevicePtr,
    pub downsample_b: DevicePtr,
    pub proj_w: DevicePtr,
    pub norm_w: DevicePtr,
    pub norm_b: DevicePtr,
    pub gate_w: DevicePtr,
    pub up_w: DevicePtr,
    pub down_w: DevicePtr,
}

Expand description

2026-09-25: Everything after the last block: the post-norm, the 2×2 conv downsample and the merger MLP whose output fills buf_out.

Fields§

§post_layernorm_w: DevicePtr

2026-09-25: Weight-only RMSNorm over hidden_size, applied before the downsample.

§downsample_w: DevicePtr

2026-09-25: The 2×2 stride-2 conv weight read as [out_hidden, hidden*4] in (in_channel, kh, kw) order, the order glm_vit_im2col_2x2 writes.

§downsample_b: DevicePtr§proj_w: DevicePtr§norm_w: DevicePtr

2026-09-25: post_projection_norm, the tower’s only LayerNorm (mean subtracted, with bias); every other norm is weight-only RMSNorm.

§norm_b: DevicePtr§gate_w: DevicePtr§up_w: DevicePtr§down_w: DevicePtr

Auto Trait Implementations§

impl Freeze for GlmVitMerger

impl RefUnwindSafe for GlmVitMerger

impl Send for GlmVitMerger

impl Sync for GlmVitMerger

impl Unpin for GlmVitMerger

impl UnwindSafe for GlmVitMerger

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T> Instrument for T

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided [Span], returning an Instrumented wrapper. Read more

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> Same for T

type Output = T

Should always be Self

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V

impl<T> WithSubscriber for T

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a [WithDispatch] wrapper. Read more

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a [WithDispatch] wrapper. Read more

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,