accelerate

7eb52556 - Fix training in DeepSpeed (#308)

Fix training in DeepSpeed (#308) * Fix training in DeepSpeed * Be more defensive * Apply suggestions from code review Co-authored-by: Zachary Mueller <muellerzr@gmail.com> Co-authored-by: Zachary Mueller <muellerzr@gmail.com>