Update zero_to_fp32.py

deepspeedai · Sep 30, 2024 · a0b2384 · a0b2384
1 parent 2cb608a
commit a0b2384
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/utils/zero_to_fp32.py b/deepspeed/utils/zero_to_fp32.py
@@ -637,7 +637,7 @@ def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None):
     parser.add_argument(
         "output_dir",
         type=str,
-        help="path to the pytorch fp32 state_dict output file "
+        help="directory to the pytorch fp32 state_dict output files"
              "(e.g. path/checkpoint-12-output/)")
     parser.add_argument(
         "--max_shard_size",