File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
torchtune/utils/_checkpointing Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -57,7 +57,7 @@ class _CheckpointerInterface(Protocol):
5757 * End-of-training Checkpointing. The model weights at the end of a completed training
5858 run are written out to file. The checkpointer ensures that the output checkpoint
5959 files have the same keys as the input checkpoint file used to begin training. The
60- checkpointer also ensures that the keys are paritioned across the same number of
60+ checkpointer also ensures that the keys are partitioned across the same number of
6161 files as the original checkpoint. This ensures that the original metadata files can
6262 be used as is, and the output checkpoint can be used with any tool that understands
6363 the original checkpoint format. This includes popular inference engines such as
@@ -460,7 +460,7 @@ def save_checkpoint(
460460 checkpoint file ``recipe_state.pt`` is created in ``_output_dir`` which contains the recipe
461461 state.
462462
463- The state_dict is first converted back to the HF format and then paritioned based on the
463+ The state_dict is first converted back to the HF format and then partitioned based on the
464464 ``_weight_map`` into separate checkpoint files.
465465
466466 Args:
You can’t perform that action at this time.
0 commit comments