From 09e151f532ab1ee796672b15c3ae9a4ce2c3e560 Mon Sep 17 00:00:00 2001 From: Alexandros Koumparoulis Date: Mon, 9 Jun 2025 16:55:31 -0700 Subject: [PATCH] use class attributes Signed-off-by: Alexandros Koumparoulis --- recipes_source/distributed_async_checkpoint_recipe.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipes_source/distributed_async_checkpoint_recipe.rst b/recipes_source/distributed_async_checkpoint_recipe.rst index a7194f6c589..5c11a9de512 100644 --- a/recipes_source/distributed_async_checkpoint_recipe.rst +++ b/recipes_source/distributed_async_checkpoint_recipe.rst @@ -74,7 +74,7 @@ Specifically: def state_dict(self): # this line automatically manages FSDP FQN's, as well as sets the default state dict type to FSDP.SHARDED_STATE_DICT - model_state_dict, optimizer_state_dict = get_state_dict(model, optimizer) + model_state_dict, optimizer_state_dict = get_state_dict(self.model, self.optimizer) return { "model": model_state_dict, "optim": optimizer_state_dict