accelerate

2106e87d - offload the previous model hook before the current module is moved to the execution device (#1315)