Chronos2#

class openstef_foundation_models.models.catalog.Chronos2(*values)[source]

Bases: StrEnum

The published Chronos-2 model sizes, each selectable as a Hub checkpoint.

The member value is the model slug, which is also the HuggingFace repo stem (OpenSTEF/<slug>-onnx) and the weights-filename stem.

BASE = 'chronos-2'
SMALL = 'chronos-2-small'
checkpoint(variant: CheckpointVariant = CheckpointVariant.DYNAMIC) → HubCheckpoint[source]

Build the Hub checkpoint reference for this size and variant.

Parameters:

  • variant (CheckpointVariant) – Which published ONNX variant to load. Defaults to the portable dynamic-shape build; pass CheckpointVariant.STATIC (or CheckpointVariant.recommended()) on macOS for CoreML.

  • variant

Returns:

A HubCheckpoint pointing at the published weights and their metadata.

Return type:

HubCheckpoint