[shardformer] fix bert and gpt downstream with new api#4024

FrankLeeeee merged 2 commits into

hpcaitech:feature/shardformerhpcaitech/ColossalAI:feature/shardformerfrom

FoolPlayer:hotfix/bert-gpt-downstreamFoolPlayer/ColossalAI:hotfix/bert-gpt-downstreamCopy head branch name to clipboard