Nevermind, I figured it out: https://github.com/facebookresearch/metaseq/blob/main/projec...