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