diff --git a/src/accelerate/accelerator.py b/src/accelerate/accelerator.py index b5d217008e1..fabd3e9fc20 100644 --- a/src/accelerate/accelerator.py +++ b/src/accelerate/accelerator.py @@ -673,7 +673,7 @@ def register_for_checkpointing(self, *objects): used in different scripts - + Every `object` must have a `load_state_dict` and `state_dict` function to be stored.