catalog#
Catalog of the foundation-model checkpoints OpenSTEF publishes.
OpenSTEF publishes each model size to its own HuggingFace repo,
OpenSTEF/<slug>-onnx, holding a few ONNX variants of the same weights. This
module mirrors that naming convention so a user selects a checkpoint by size and
variant instead of hand-typing repo ids and filenames. The strings here are the
wire contract with the publisher (openstef-checkpoints); keeping them in one
place is what lets the two repos stay in step.
Two variants matter when selecting:
CheckpointVariant.DYNAMIC— symbolic shapes; runs on every execution provider and is the portable default.CheckpointVariant.STATIC— frozen shapes; on macOS this is what lets the CoreML provider engage in the default fallback chain, so prefer it there.
The module is import-light (pure pydantic config, no inference runtime), so a checkpoint can be selected and a config built without ONNX Runtime installed.
Example:
from openstef_foundation_models.models.catalog import Chronos2, CheckpointVariant
checkpoint = Chronos2.BASE.checkpoint(CheckpointVariant.STATIC)
Classes#
|
A published ONNX variant of a model's weights. |
|
The published Chronos-2 model sizes, each selectable as a Hub checkpoint. |