Skip to content

remove bad uma checkpoint and make task heads mandatory to specify - #219

Merged
WillEngler merged 1 commit into
mainfrom
will/remove-bad-uma
Aug 21, 2026
Merged

remove bad uma checkpoint and make task heads mandatory to specify#219
WillEngler merged 1 commit into
mainfrom
will/remove-bad-uma

Conversation

@WillEngler

Copy link
Copy Markdown
Contributor

Changes driven by Garden-AI/almanac#43

@WillEngler
WillEngler merged commit 2664674 into main Aug 21, 2026
5 checks passed
@WillEngler
WillEngler deleted the will/remove-bad-uma branch August 21, 2026 15:34
OwenPriceSkelly added a commit that referenced this pull request Sep 1, 2026
The download phase runs setup(checkpoint, "cpu") via _run_download, but
unlike verify_checkpoint it never fell back to the env's declared
VERIFY_KWARGS when no explicit setup_kwargs were given. Since #219 made
multi-head checkpoints raise without a head selection, every such
checkpoint (uma task=..., mace-mh-1 head=...) failed the sync's download
phase with e.g. "download: ValueError: uma-s-1p1 is multi-task and has
no default head" (first seen on the 2026-08-28 Aurora sync, 1.6.3).

Resolve the hosting env first, then default setup_kwargs to
verify_kwargs_for(root, env_name, checkpoint) — explicit kwargs still
win, and envs without VERIFY_KWARGS still get {}.

Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant