SmolLm2Config

Configuration values used by a SmolLM2 causal language model.

Record fields

NameDescription
BosTokenId
EosTokenId
HiddenSize
IntermediateSize
MaxPositionEmbeddings
NumAttentionHeads
NumHiddenLayers
NumKeyValueHeads
RmsNormEps
RopeTheta
VocabSize

SmolLm2Config module

Validation operations for SmolLM2 configurations.

FunctionDescription
validate configValidate dimensions and constants required by the implemented architecture.

validate

validate config

Validate dimensions and constants required by the implemented architecture.

Parameters

  • config : SmolLm2Config

SmolLm2Input

Tensor inputs accepted by SmolLM2.

SmolLm2Output

Tensor outputs produced by SmolLM2.