Skip to main content

ModelConfig

Trait ModelConfig 

Source
pub trait ModelConfig:
    Send
    + Sync
    + Debug {
    // Required methods
    fn architecture(&self) -> &str;
    fn vocab_size(&self) -> usize;
    fn hidden_size(&self) -> usize;
    fn num_layers(&self) -> usize;
    fn validate(&self) -> Result<()>;
}
Expand description

Model configuration trait

Required Methods§

Source

fn architecture(&self) -> &str

Get model architecture name

Source

fn vocab_size(&self) -> usize

Get vocabulary size

Source

fn hidden_size(&self) -> usize

Get hidden dimension

Source

fn num_layers(&self) -> usize

Get number of layers

Source

fn validate(&self) -> Result<()>

Validate configuration

Dyn Compatibility§

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors§

Source§

impl ModelConfig for ArcticConfig

Source§

impl ModelConfig for BLOOMConfig

Source§

impl ModelConfig for BaichuanConfig

Source§

impl ModelConfig for BertConfig

Source§

impl ModelConfig for CLIPConfig

Source§

impl ModelConfig for ChatGLMConfig

Source§

impl ModelConfig for CodeLlamaConfig

Source§

impl ModelConfig for CogVLMConfig

Source§

impl ModelConfig for DbrxConfig

Source§

impl ModelConfig for DeepSeekConfig

Source§

impl ModelConfig for DeepSeekMoEConfig

Source§

impl ModelConfig for EncodecConfig

Source§

impl ModelConfig for FalconConfig

Source§

impl ModelConfig for FlorenceConfig

Source§

impl ModelConfig for GPT2Config

Source§

impl ModelConfig for GPTJConfig

Source§

impl ModelConfig for GPTNeoXConfig

Source§

impl ModelConfig for GemmaConfig

Source§

impl ModelConfig for GraniteConfig

Source§

impl ModelConfig for GrokConfig

Source§

impl ModelConfig for HuBERTConfig

Source§

impl ModelConfig for IdeficsConfig

Source§

impl ModelConfig for InternLMConfig

Source§

impl ModelConfig for InternVLConfig

Source§

impl ModelConfig for JambaConfig

Source§

impl ModelConfig for LLaVAConfig

Source§

impl ModelConfig for LlamaConfig

Source§

impl ModelConfig for MPTConfig

Source§

impl ModelConfig for MambaConfig

Source§

impl ModelConfig for MiniCPMConfig

Source§

impl ModelConfig for MistralConfig

Source§

impl ModelConfig for MixtralConfig

Source§

impl ModelConfig for MusicGenConfig

Source§

impl ModelConfig for OPTConfig

Source§

impl ModelConfig for OlmoConfig

Source§

impl ModelConfig for Phi3VisionConfig

Source§

impl ModelConfig for PhiConfig

Source§

impl ModelConfig for Qwen2VLConfig

Source§

impl ModelConfig for QwenConfig

Source§

impl ModelConfig for RecurrentGemmaConfig

Source§

impl ModelConfig for Rwkv4Config

Source§

impl ModelConfig for Rwkv6Config

Source§

impl ModelConfig for StarCoderConfig

Source§

impl ModelConfig for T5Config

Source§

impl ModelConfig for Wav2Vec2Config

Source§

impl ModelConfig for WhisperConfig

Source§

impl ModelConfig for YiConfig