accelerate

latest bnb no longer has optim_args attribute on optimizer

#3311

Merged