update fsdp docs (#2026)

huggingface · Oct 3, 2023 · 76ee7f2 · 76ee7f2
1 parent 420743a
commit 76ee7f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/usage_guides/fsdp.md b/docs/source/usage_guides/fsdp.md
@@ -154,7 +154,7 @@ When using transformers `save_pretrained`, pass `state_dict=accelerator.get_stat
       args.output_dir,
       is_main_process=accelerator.is_main_process,
       save_function=accelerator.save,
-+     state_dict=accelerator.get_state_dict(model),
++     state_dict=accelerator.get_state_dict(model, unwrap=False),
 )
 ```