transformer_lens.tools.model_registry.checkpoints module

Training-checkpoint label data for checkpointed model families.

Canonical home for the checkpoint schedules previously defined in transformer_lens/loading_from_pretrained.py. The schedules are frozen historical artifacts of the published training runs.

transformer_lens.tools.model_registry.checkpoints.get_checkpoint_labels(model_name: str) → tuple[list[int], str]

Return (checkpoint labels, label type) for a checkpointed model family.

Covers the HF-revision-checkpointed families (Pythia, stanford-crfm). Raises ValueError for models without published checkpoint schedules.