Files
ColossalAI/examples/tutorial/handson6/scripts/inpaint.py