Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::LlamaConfig Member List

This is the complete list of members for Mila::Dnn::LlamaConfig, including all inherited members.

embedding_dim_Mila::Dnn::LlamaConfigprivate
fromMetadata(const SerializationMetadata &meta)Mila::Dnn::LlamaConfiginlinevirtual
getHiddenDimension() const noexceptMila::Dnn::LlamaConfiginline
getMaxSequenceLength() const noexceptMila::Dnn::LlamaConfiginline
getModelDim() const noexceptMila::Dnn::LlamaConfiginline
getNumHeads() const noexceptMila::Dnn::LlamaConfiginline
getNumKVHeads() const noexceptMila::Dnn::LlamaConfiginline
getNumLayers() const noexceptMila::Dnn::LlamaConfiginline
getRMSNormEpsilon() const noexceptMila::Dnn::LlamaConfiginline
getRoPEScalingFactor() const noexceptMila::Dnn::LlamaConfiginline
getRoPETheta() const noexceptMila::Dnn::LlamaConfiginline
getVocabSize() const noexceptMila::Dnn::LlamaConfiginline
hidden_dim_Mila::Dnn::LlamaConfigprivate
LlamaConfig(dim_t embedding_dim, dim_t num_layers)Mila::Dnn::LlamaConfiginline
max_seq_len_Mila::Dnn::LlamaConfigprivate
num_heads_Mila::Dnn::LlamaConfigprivate
num_kv_heads_Mila::Dnn::LlamaConfigprivate
num_layers_Mila::Dnn::LlamaConfigprivate
rms_norm_eps_Mila::Dnn::LlamaConfigprivate
rope_scaling_factor_Mila::Dnn::LlamaConfigprivate
rope_theta_Mila::Dnn::LlamaConfigprivate
toMetadata() constMila::Dnn::LlamaConfiginlinevirtual
toString() const overrideMila::Dnn::LlamaConfiginlinevirtual
use_bias_Mila::Dnn::LlamaConfigprivate
useBias() const noexceptMila::Dnn::LlamaConfiginline
validate() const overrideMila::Dnn::LlamaConfiginlinevirtual
vocab_size_Mila::Dnn::LlamaConfigprivate
withBias(this Self &&self, bool use_bias)Mila::Dnn::LlamaConfiginline
withHiddenDimension(this Self &&self, dim_t hidden_dim)Mila::Dnn::LlamaConfiginline
withMaxSequenceLength(this Self &&self, dim_t max_seq_len)Mila::Dnn::LlamaConfiginline
withNumHeads(this Self &&self, dim_t num_heads)Mila::Dnn::LlamaConfiginline
withNumKVHeads(this Self &&self, dim_t num_kv_heads)Mila::Dnn::LlamaConfiginline
withRoPEScalingFactor(this Self &&self, float scale_factor)Mila::Dnn::LlamaConfiginline
withRoPETheta(this Self &&self, float theta)Mila::Dnn::LlamaConfiginline
withVocabularyLength(this Self &&self, dim_t vocab_size)Mila::Dnn::LlamaConfiginline
~ComponentConfig()=defaultMila::Dnn::ComponentConfigvirtual