Megatron-DeepSpeed

aa706154 - Add temporary assert to finetuning until it can be fixed.