prepare deepspeed accomodate fp16 and bf16#1728

vwxyzjn merged 2 commits intohuggingface:mainhuggingface/trl:mainfrom

mnoukhov:prepare-ds-fpmnoukhov/trl:prepare-ds-fpCopy head branch name to clipboard