[BUG]: Failed to trace GPT2Model

[TXacs]

🐛 Describe the bug

when I run the script ColossalAI/tests/test_fx/test_pipeline/test_hf_model/test_gpt.py, got this error.

env: torch==2.1.2 transformers==4.32.0 colossalai==0.3.4

Environment

No response