CheckpointStateSaveConfigBase¶

Module: fast_llm.engine.checkpoint.config

Inherits from: CheckpointSaveConfigBase, CheckpointStateConfigBase, CheckpointConfigBase

Fields¶

format — core

Type: type Default: FastLLMCheckpointFormat

Format of the checkpoint.

data_type — feature

Type: DataType or None Default: None

Data type to save the checkpoint.

model_weights — feature

Type: bool Default: True

Save the model weights.

optimizer_state — feature

Type: bool Default: None

Save the optimizer state. Default: save if supported by the format.

parameters_per_file — feature

Type: int Default: 4_294_967_296

Limit the number of parameters saved in each file.