Accelerator.save_state errors out due to timeout. Unable to increase timeout through kwargs_handlers

Maybe ongoing issue.