Struct BertConfig 

pub struct BertConfig {
    pub vocab_size: usize,
    pub hidden_size: usize,
    pub num_hidden_layers: usize,
    pub num_attention_heads: usize,
    pub intermediate_size: usize,
    pub max_position_embeddings: usize,
    pub type_vocab_size: usize,
    pub layer_norm_eps: f64,
    pub hidden_act: String,
}

BERT model configuration.

Fields

vocab_size: usize

hidden_size: usize

num_hidden_layers: usize

num_attention_heads: usize

intermediate_size: usize

max_position_embeddings: usize

type_vocab_size: usize

layer_norm_eps: f64

hidden_act: String

Implementations

impl BertConfig

pub fn from_file(path: &Path) -> Result<BertConfig, Error>

pub fn head_dim(&self) -> usize

impl BertConfig

pub fn from_gguf(raw: &GgufFile) -> Result<BertConfig, Error>

Trait Implementations

impl Clone for BertConfig

fn clone(&self) -> BertConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for BertConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for BertConfig

fn deserialize<__D>( __deserializer: __D, ) -> Result<BertConfig, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.