Skip to content

Commit

Permalink
Stop writing msgpack file for new checkpoints and update empty nodes …
Browse files Browse the repository at this point in the history 
…handling so that it no longer depends on this file.

PiperOrigin-RevId: 636665054
  • Loading branch information
@cpgaffney1
cpgaffney1 authored and Flax Authors committed Jun 28, 2024
1 parent 15e0e8d commit b8a805b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flax/training/checkpoints.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@
COMMIT_SUCCESS_FILE = 'commit_success.txt'

# Orbax main checkpoint file name.
ORBAX_CKPT_FILENAME = 'checkpoint'
ORBAX_CKPT_FILENAME = '_METADATA'
ORBAX_MANIFEST_OCDBT = 'manifest.ocdbt'

PyTree = Any
Expand Down

0 comments on commit b8a805b

Please sign in to comment.