Skip to content

Commit

Permalink
Fix typo in monolithic chkpt callback docs (#1024)
Browse files Browse the repository at this point in the history
* Fix typo in monolithic chkpt callback docs

* reorder to match function signature
  • Loading branch information
sashaDoubov committed Mar 13, 2024
1 parent 257c25d commit 4e8a875
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions llmfoundry/callbacks/monolithic_ckpt_callback.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,10 @@ class MonolithicCheckpointSaver(Callback):
Args:
save_folder (str): Folder to save checkpoints to (can be a URI)
filename (str): Filename to save checkpoints to.
batch_interval (int): Number of batches between checkpoints.
filename (str): Filename to save checkpoints to.
overwrite (bool): Whether to overwrite previous checkpoints.
keep_optimizer(bool): Whether to save the optimizer state in the monolithic checkpoint.
keep_optimizers (bool): Whether to save the optimizer state in the monolithic checkpoint.
"""

def __init__(self,
Expand Down

0 comments on commit 4e8a875

Please sign in to comment.