Files
ColossalAI/examples/language/opt/context.py