Megatron-DeepSpeed

8044c7b4 - Update arguments checks.

Update arguments checks. hidden_size % attention_heads == 0 is handled above when dealing with kv_channels. Adding check for decoder sequence length.