Update the DeepSpeed Phi-2 impl. to work with the HF latest changes (#4950)
The latest changes in Huggingface Phi-2 implementation
(https://huggingface.co/microsoft/phi-2/commit/cb2f4533604d8b67de604e7df03bfe6f3ca22869#d2h-025836)
have broken the DeepSpeed implementation. This PR address the related
issues.
---------
Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>