pub struct Config {Show 26 fields
pub grid: usize,
pub window: usize,
pub vit_dim: usize,
pub vit_heads: usize,
pub vit_layers: usize,
pub vit_ffn_pad: usize,
pub vit_global: Vec<usize>,
pub patch: usize,
pub image_size: usize,
pub d_model: usize,
pub d_heads: usize,
pub d_layers: usize,
pub d_ffn: usize,
pub text_len: usize,
pub text_dim: usize,
pub text_heads: usize,
pub text_layers: usize,
pub text_ffn: usize,
pub text_eps: f32,
pub vit_eps: f32,
pub queries: usize,
pub dec_layers: usize,
pub neck_splits: Vec<usize>,
pub mask_size: usize,
pub vit_device: bool,
pub vit_res_f32: bool,
}Expand description
Model constants (from the manifest’s param lines).
Fields§
§grid: usize§window: usize§vit_dim: usize§vit_heads: usize§vit_layers: usize§vit_ffn_pad: usize§vit_global: Vec<usize>§patch: usize§image_size: usize§d_model: usize§d_heads: usize§d_layers: usize§d_ffn: usize§text_len: usize§text_dim: usize§text_heads: usize§text_layers: usize§text_ffn: usize§text_eps: f32§vit_eps: f32§queries: usize§dec_layers: usize§neck_splits: Vec<usize>§mask_size: usize§vit_device: boolevery ViT layer on the device (RoPE, LayerNorms, residual adds); bundles from before it host that glue
vit_res_f32: boolthe device ViT’s residual stream in f32 (AddLayerNorm(f32_residual));
bundles from before it keep it bf16
Implementations§
Trait Implementations§
Auto Trait Implementations§
impl Freeze for Config
impl RefUnwindSafe for Config
impl Send for Config
impl Sync for Config
impl Unpin for Config
impl UnsafeUnpin for Config
impl UnwindSafe for Config
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