Move import to when needed to avoid circular dependency error (#19579)
### Description
Move import to when needed to avoid circular dependency error
### Motivation and Context
Fixes dependency error described here:
https://github.com/microsoft/DeepSpeed/issues/5140
---------
Co-authored-by: Thiago Crepaldi <thiago.crepaldi@microsoft.com>