pub struct TritTransformerConfig {
pub dim: usize,
pub n_layers: usize,
pub n_heads: usize,
pub n_kv_heads: usize,
pub vocab_size: usize,
pub multiple_of: usize,
pub ffn_dim_multiplier: Option<f64>,
pub norm_eps: f32,
pub max_seq_len: usize,
}Fields§
§dim: usize§n_layers: usize§n_heads: usize§n_kv_heads: usize§vocab_size: usize§multiple_of: usize§ffn_dim_multiplier: Option<f64>§norm_eps: f32§max_seq_len: usizeTrait Implementations§
Auto Trait Implementations§
impl Freeze for TritTransformerConfig
impl RefUnwindSafe for TritTransformerConfig
impl Send for TritTransformerConfig
impl Sync for TritTransformerConfig
impl Unpin for TritTransformerConfig
impl UnsafeUnpin for TritTransformerConfig
impl UnwindSafe for TritTransformerConfig
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
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more