Struct aiha::models::bloom::BloomModelConfig

pub struct BloomModelConfig { /* private fields */ }

A struct representing a BLOOM model configuration

Implementations

impl BloomModelConfig

BLOOM model implementation

pub fn new( params: BloomParams, model_type: String, available_libraries: Vec<ModelLibraries> ) -> BloomModelConfig

Build a new BloomModelConfig struct based on the provided parameters

Trait Implementations

impl Clone for BloomModelConfig

fn clone(&self) -> BloomModelConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for BloomModelConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for BloomModelConfig

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

Deserialize this value from the given Serde deserializer.

impl ModelConfigTrait for BloomModelConfig

Implementation of the ModelConfigTrait trait for BloomModelConfig

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

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

Create a new model config from a JSON value.