Accelerate: 0.9.0 Python: 3.9.7 I have written the following class method that saves off a model trained when using accelerate: def save_model(self, model_to_save: torch.nn.Module, model_save_path: str): if self.is_accelerate and self.accelerator.is_local_main_process: self.…

[image] aclifton314: I have written the following class method that saves off a model trained when using accelerate: Here’s what you should really do instead to make sure it all works well. This is how we have Accelerator.save_state currently as well: def save_model(self, accelerator, model_…

Error when saving model in accelerate

🤗Accelerate

aclifton314 July 6, 2022, 11:36pm 6

That worked perfectly. Thanks!

Topic		Replies	Views
Accelerator.save_state errors out due to timeout. Unable to increase timeout through kwargs_handlers 🤗Accelerate	5	1563	March 3, 2025
NCCL Timeout Accelerate Load From Checkpoint 🤗Accelerate	2	2731	June 20, 2025
NCCL Watchdog Timeout error while using Deepspeed and accelerate Beginners	13	1569	November 5, 2025
Code RuntimeError 🤗Accelerate	2	1408	October 22, 2023
Error occurs when saving model in multi-gpu settings Beginners	4	1514	November 9, 2021

Error when saving model in accelerate

Related topics