DeepSpeed

85334238 - added check to avoid undefined behavior when the input_id length is greater than max_tokens (#4349)