Trait ModelConfigTrait 

pub trait ModelConfigTrait {
    // Required method
    fn from_json(value: Value) -> Result<Self, ModelError>
       where Self: Sized;

    // Provided methods
    fn hidden_size(&self) -> i32 { ... }
    fn intermediate_size(&self) -> i32 { ... }
    fn max_position_embeddings(&self) -> i32 { ... }
    fn num_attention_heads(&self) -> i32 { ... }
    fn num_hidden_layers(&self) -> i32 { ... }
    fn model_type(&self) -> &str { ... }
    fn available_libraries(&self) -> &[ModelLibraries] { ... }
}

Generic trait for Hugging Face models

Required Methods

fn from_json(value: Value) -> Result<Self, ModelError>

where Self: Sized,

Create a new model config from a JSON value.

Provided Methods

fn hidden_size(&self) -> i32

Returns the model hidden size

fn intermediate_size(&self) -> i32

Returns the model intermediate size

fn max_position_embeddings(&self) -> i32

Returns the model max position embeddings

fn num_attention_heads(&self) -> i32

Returns the model number of attention heads

fn num_hidden_layers(&self) -> i32

Returns the model number of hidden layers

fn model_type(&self) -> &str

Returns the model type

fn available_libraries(&self) -> &[ModelLibraries]

Returns the model libraries

Implementors

impl ModelConfigTrait for ModelConfig

Model config implementation

impl ModelConfigTrait for BertModelConfig

Implementation of the ModelConfigTrait trait for BertModelConfig

impl ModelConfigTrait for BloomModelConfig

Implementation of the ModelConfigTrait trait for BloomModelConfig

impl ModelConfigTrait for GPT2ModelConfig

Implementation of the ModelConfigTrait trait for GPT2ModelConfig

impl ModelConfigTrait for GPTJModelConfig

Implementation of the ModelConfigTrait trait for GPTJModelConfig

impl ModelConfigTrait for GPTNeoModelConfig

Implementation of the ModelConfigTrait trait for GPTNeoModelConfig

impl ModelConfigTrait for LlamaModelConfig

Implementation of the ModelConfigTrait trait for LlamaModelConfig

impl ModelConfigTrait for OPTModelConfig

Implementation of the ModelConfigTrait trait for OPTModelConfig

impl ModelConfigTrait for T5ModelConfig

Implementation of the ModelConfigTrait trait for T5ModelConfig