fix integration

[rui-ren]

What does this PR do?

Fixes # (issue)

File "/opt/conda/envs/ptca/lib/python3.8/site-packages/optimum/onnxruntime/trainer.py", line 58, in <module>
    from transformers.deepspeed import deepspeed_init, deepspeed_load_checkpoint, is_deepspeed_zero3_enabled
ImportError: cannot import name 'deepspeed_load_checkpoint' from 'transformers.deepspeed'

Issue

Transformers moved 3rd party package to integrations folder, this PR is to make this file compatible with transformers>4.31.0

huggingface/transformers@4b79697

Before submitting

• This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
• Did you make sure to update the documentation with your changes?
• Did you write any new necessary tests?