[shardformer]import transformers implicitly

[FrankLeeeee]

As Hugging Face Transformers is not our dependency, we should not import from transformers at the top of our files. We should only import them inside the function when needed.

[FrankLeeeee]

Fixed by #4101