[deepspeed] replaced deprecated init arg
deepspeed 0.5.2 deprecated `config` in `zero.Init()` in favor of `config_dict_or_path` - deepspeed 0.5.3 fixed a bug in the previous version related to that https://github.com/microsoft/DeepSpeed/pull/1373. So this PR updates the arg and ups the minimal requirement.

0.5.3 fixes a bug in `zero_to_fp32.py` introduced by 0.5.2

The ci failure is unrelated.

@sgugger, @LysandreJik 

