Commit 60deff4
evo2_megatron: load Savanna HF checkpoints with weights_only=False
torch 2.6 changed the default of `weights_only` to True. The Savanna
checkpoint pickle includes numpy globals (`numpy.core.multiarray._reconstruct`),
which the safer loader rejects. The converter then exits 0 with no output
written and the error gets buried in stderr — silent failure.
The Savanna repos under arcinstitute/* are trusted sources, so load with
weights_only=False.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>1 parent b0953dc commit 60deff4
1 file changed
Lines changed: 1 addition & 1 deletion
File tree
- bionemo-recipes/recipes/evo2_megatron/src/bionemo/evo2/utils/checkpoint
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
135 | 135 | | |
136 | 136 | | |
137 | 137 | | |
138 | | - | |
| 138 | + | |
139 | 139 | | |
140 | 140 | | |
141 | 141 | | |
| |||
0 commit comments