pub struct LlamaConfig {Show 13 fields
pub hidden_size: usize,
pub intermediate_size: usize,
pub vocab_size: usize,
pub num_hidden_layers: usize,
pub num_attention_heads: usize,
pub num_key_value_heads: Option<usize>,
pub rms_norm_eps: f64,
pub rope_theta: f32,
pub bos_token_id: Option<u32>,
pub eos_token_id: Option<LlamaEosToks>,
pub rope_scaling: Option<Llama3RopeConfig>,
pub max_position_embeddings: usize,
pub tie_word_embeddings: Option<bool>,
}Fields§
§intermediate_size: usize§vocab_size: usize§num_attention_heads: usize§num_key_value_heads: Option<usize>§rms_norm_eps: f64§rope_theta: f32§bos_token_id: Option<u32>§eos_token_id: Option<LlamaEosToks>§rope_scaling: Option<Llama3RopeConfig>§max_position_embeddings: usize§tie_word_embeddings: Option<bool>Implementations§
Source§impl LlamaConfig
impl LlamaConfig
pub fn num_key_value_heads(&self) -> usize
Source§impl LlamaConfig
impl LlamaConfig
pub fn into_config(self, use_flash_attn: bool) -> Config
Trait Implementations§
Source§impl Clone for LlamaConfig
impl Clone for LlamaConfig
Source§fn clone(&self) -> LlamaConfig
fn clone(&self) -> LlamaConfig
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for LlamaConfig
impl Debug for LlamaConfig
Source§impl<'de> Deserialize<'de> for LlamaConfig
impl<'de> Deserialize<'de> for LlamaConfig
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for LlamaConfig
impl RefUnwindSafe for LlamaConfig
impl Send for LlamaConfig
impl Sync for LlamaConfig
impl Unpin for LlamaConfig
impl UnwindSafe for LlamaConfig
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> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
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