Accelerate version errors in Trainer

[muellerzr]

@BBIBIBBIBI

The accelerate version seems to change midway through, is there a workaround?

If so, you need to restart your session so the new libraries can be loaded in