Skip to content

Commit

Permalink
fix dcp tutorial formatting (#2990)
Browse files Browse the repository at this point in the history
  • Loading branch information
H-Huang authored Aug 1, 2024
1 parent fda8184 commit 740e3b0
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions recipes_source/distributed_checkpoint_recipe.rst
Original file line number Diff line number Diff line change
Expand Up @@ -333,6 +333,7 @@ Since this can be an issue when users wish to share models with users used to th
to their applications. For this case, we provide the ``format_utils`` module in ``torch.distributed.checkpoint.format_utils``.

A command line utility is provided for the users convenience, which follows the following format:

.. code-block:: bash
python -m torch.distributed.checkpoint.format_utils -m <checkpoint location> <location to write formats to> <mode>
Expand All @@ -341,6 +342,7 @@ In the command above, ``mode`` is one of ``torch_to_dcp``` or ``dcp_to_torch``.


Alternatively, methods are also provided for users who may wish to convert checkpoints directly.

.. code-block:: python
import os
Expand Down

0 comments on commit 740e3b0

Please sign in to comment.