Skip to content

Commit

Permalink
Fix typos (pytorch#1086)
Browse files Browse the repository at this point in the history
  • Loading branch information
mdeff authored Jun 13, 2024
1 parent 12cbc4b commit 1759a23
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torchtune/utils/_checkpointing/_checkpointer.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ class _CheckpointerInterface(Protocol):
* End-of-training Checkpointing. The model weights at the end of a completed training
run are written out to file. The checkpointer ensures that the output checkpoint
files have the same keys as the input checkpoint file used to begin training. The
checkpointer also ensures that the keys are paritioned across the same number of
checkpointer also ensures that the keys are partitioned across the same number of
files as the original checkpoint. This ensures that the original metadata files can
be used as is, and the output checkpoint can be used with any tool that understands
the original checkpoint format. This includes popular inference engines such as
Expand Down Expand Up @@ -460,7 +460,7 @@ def save_checkpoint(
checkpoint file ``recipe_state.pt`` is created in ``_output_dir`` which contains the recipe
state.
The state_dict is first converted back to the HF format and then paritioned based on the
The state_dict is first converted back to the HF format and then partitioned based on the
``_weight_map`` into separate checkpoint files.
Args:
Expand Down

0 comments on commit 1759a23

Please sign in to comment.