Due to size limitation, we provide only the code without checkpoints for reference.

Training command:
