pub enum ModelType {
Llama,
Mistral,
Phi,
Qwen,
Gemma,
Unknown,
}Expand description
Supported model architectures
Variants§
Llama
Llama family (Llama 2, Llama 3, etc.)
Mistral
Mistral models
Phi
Microsoft Phi models
Qwen
Qwen models (Alibaba)
Gemma
Google Gemma models
Unknown
Unknown/unsupported
Implementations§
Source§impl ModelType
impl ModelType
Sourcepub fn from_config(config: &ModelConfig) -> Self
pub fn from_config(config: &ModelConfig) -> Self
Detect model type from config.
Sourcepub fn is_supported(&self) -> bool
pub fn is_supported(&self) -> bool
Check if the model type is supported by Ferrotensor.
Trait Implementations§
impl Copy for ModelType
impl Eq for ModelType
impl StructuralPartialEq for ModelType
Auto Trait Implementations§
impl Freeze for ModelType
impl RefUnwindSafe for ModelType
impl Send for ModelType
impl Sync for ModelType
impl Unpin for ModelType
impl UnsafeUnpin for ModelType
impl UnwindSafe for ModelType
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