mstar.model.waypoint.checkpoint#

Resolve and validate the two checkpoints used by Waypoint.

Resolution happens before model construction so a bad repository, partial local download, or incompatible config.yaml cannot allocate several GiB of GPU storage before failing. Local paths always win and never touch Hugging Face.

Functions

require_taehv_runtime()

Fail before checkpoint download or module allocation if TAEHV is absent.

resolve_taehv_checkpoint(source, *[, ...])

Resolve only taehv1_5.pth from a local path or HF repository.

resolve_waypoint_checkpoint(source, config, *)

Resolve a local directory or HF repo ID and validate it before allocation.

validate_waypoint_checkpoint(checkpoint_dir, ...)

Validate files and checkpoint facts, returning an absolute directory.

mstar.model.waypoint.checkpoint.require_taehv_runtime()[source]#

Fail before checkpoint download or module allocation if TAEHV is absent.

Return type:

None

mstar.model.waypoint.checkpoint.resolve_taehv_checkpoint(source, *, cache_dir=None, revision=None)[source]#

Resolve only taehv1_5.pth from a local path or HF repository.

Parameters:
Return type:

Path

mstar.model.waypoint.checkpoint.resolve_waypoint_checkpoint(source, config, *, cache_dir=None, revision=None)[source]#

Resolve a local directory or HF repo ID and validate it before allocation.

Parameters:
Return type:

Path

mstar.model.waypoint.checkpoint.validate_waypoint_checkpoint(checkpoint_dir, config)[source]#

Validate files and checkpoint facts, returning an absolute directory.

Parameters:
Return type:

Path