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