Skip to content

GPTBaseModelConfig

Module: fast_llm.models.gpt.config

Inherits from: LanguageModelConfig, BlockConfig, BaseModelConfig

Fields

decoderarchitecture

Type: BlockSequenceConfig    Default: (sub-fields optional)

Configuration for the language model decoder.

embeddingsarchitecture

Type: LanguageModelEmbeddingsConfig    Default: (sub-fields optional)

Configuration for the language model embeddings.

headarchitecture

Type: LanguageModelHeadConfig    Default: (sub-fields optional)

Configuration for the language model head(s).

hidden_sizearchitecture

Type: int    Default: 1024

Size of the model's main hidden dimension, e.g., for its input and output layers.

peftarchitecture

Type: PeftConfig    Default: (sub-fields optional)

Configuration for parameter-efficient fine tuning.

tied_embedding_weightarchitecture

Type: bool    Default: False

Tie the output weights (logits) with the vocabulary embedding.

lr_scalefeature

Type: float or None    Default: None

Scaling factor for the layer learning rate. Combines multiplicatively with the scale set by the parent and child layers, if applicable.

Used in