pub struct YiConfig {Show 17 fields
pub vocab_size: usize,
pub hidden_size: usize,
pub intermediate_size: usize,
pub num_hidden_layers: usize,
pub num_attention_heads: usize,
pub num_key_value_heads: usize,
pub hidden_act: String,
pub max_position_embeddings: usize,
pub initializer_range: f32,
pub rms_norm_eps: f32,
pub use_cache: bool,
pub pad_token_id: i64,
pub bos_token_id: i64,
pub eos_token_id: i64,
pub tie_word_embeddings: bool,
pub rope_theta: f32,
pub attention_dropout: f32,
}Fields§
§vocab_size: usize§intermediate_size: usize§num_attention_heads: usize§num_key_value_heads: usize§max_position_embeddings: usize§initializer_range: f32§rms_norm_eps: f32§use_cache: bool§pad_token_id: i64§bos_token_id: i64§eos_token_id: i64§tie_word_embeddings: bool§rope_theta: f32§attention_dropout: f32Implementations§
Source§impl YiConfig
impl YiConfig
Sourcepub fn from_gguf_config(gguf: &GGUFModelConfig) -> Self
pub fn from_gguf_config(gguf: &GGUFModelConfig) -> Self
Create YiConfig from GGUF model configuration
Trait Implementations§
Source§impl<'de> Deserialize<'de> for YiConfig
impl<'de> Deserialize<'de> for YiConfig
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
Source§impl ModelConfig for YiConfig
impl ModelConfig for YiConfig
Auto Trait Implementations§
impl Freeze for YiConfig
impl RefUnwindSafe for YiConfig
impl Send for YiConfig
impl Sync for YiConfig
impl Unpin for YiConfig
impl UnsafeUnpin for YiConfig
impl UnwindSafe for YiConfig
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> DeserializeOwned for Twhere
T: for<'de> Deserialize<'de>,
impl<T> ErasedDestructor for Twhere
T: 'static,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§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