[shardformer] adapted llama to the new API#4036

FrankLeeeee merged 1 commit intohpcaitech:feature/shardformerhpcaitech/ColossalAI:feature/shardformerfrom

FrankLeeeee:hotfix/llama-t5-fixFrankLeeeee/ColossalAI:hotfix/llama-t5-fixCopy head branch name to clipboard