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

I checked this with different timeout=timedelta(seconds=TK) value ranging from 1 to 10. The code never fails for any of these values.