From ff36e59d3f9e5ee9391127ff76db0853f855cd30 Mon Sep 17 00:00:00 2001 From: Shoufa Chen Date: Wed, 7 May 2025 15:07:49 -0700 Subject: [PATCH] Update distributed_checkpoint_recipe.rst --- recipes_source/distributed_checkpoint_recipe.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipes_source/distributed_checkpoint_recipe.rst b/recipes_source/distributed_checkpoint_recipe.rst index 950839966c5..8a81d63bb65 100644 --- a/recipes_source/distributed_checkpoint_recipe.rst +++ b/recipes_source/distributed_checkpoint_recipe.rst @@ -352,7 +352,7 @@ Alternatively, methods are also provided for users who may wish to convert check dcp_to_torch_save(CHECKPOINT_DIR, TORCH_SAVE_CHECKPOINT_DIR) # converts the torch.save model back to DCP - dcp_to_torch_save(TORCH_SAVE_CHECKPOINT_DIR, f"{CHECKPOINT_DIR}_new") + torch_save_to_dcp(TORCH_SAVE_CHECKPOINT_DIR, f"{CHECKPOINT_DIR}_new")