pub struct ModelCfg {Show 31 fields
pub dim: usize,
pub hc: usize,
pub iters: usize,
pub hc_eps: f32,
pub eps: f32,
pub n_layers: usize,
pub vocab: usize,
pub max_seq: usize,
pub n_heads: usize,
pub head_dim: usize,
pub rope_dim: usize,
pub q_rank: usize,
pub o_rank: usize,
pub groups: usize,
pub window: usize,
pub ratios: Vec<usize>,
pub kv_sources: Vec<usize>,
pub index_sources: Vec<usize>,
pub cand_src: i64,
pub cand_topk_blocks: usize,
pub cand_block: usize,
pub index_heads: usize,
pub index_hd: usize,
pub index_topk: usize,
pub rope_theta: f32,
pub compress_rope_theta: f32,
pub rope_factor: f32,
pub orig_seq: usize,
pub beta_fast: f32,
pub beta_slow: f32,
pub moe: MoeCfg,
}Fields§
§dim: usize§hc: usize§iters: usize§hc_eps: f32§eps: f32§n_layers: usize§vocab: usize§max_seq: usize§n_heads: usize§head_dim: usize§rope_dim: usize§q_rank: usize§o_rank: usize§groups: usize§window: usize§ratios: Vec<usize>§kv_sources: Vec<usize>§index_sources: Vec<usize>§cand_src: i64§cand_topk_blocks: usize§cand_block: usize§index_heads: usize§index_hd: usize§index_topk: usize§rope_theta: f32§compress_rope_theta: f32§rope_factor: f32§orig_seq: usize§beta_fast: f32§beta_slow: f32§moe: MoeCfgImplementations§
Source§impl ModelCfg
impl ModelCfg
pub fn from_golden(g: &Golden) -> Self
pub fn attn_cfg(&self, layer: usize) -> CompAttnCfg
pub fn indexer_cfg(&self) -> IndexerCfg
Auto Trait Implementations§
impl Freeze for ModelCfg
impl RefUnwindSafe for ModelCfg
impl Send for ModelCfg
impl Sync for ModelCfg
impl Unpin for ModelCfg
impl UnwindSafe for ModelCfg
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