https://github.com/LSimon95/megatts2/blob/c9ca2a88febf9db2cf4d8da0860efc9948db2b76/modules/mrte.py#L63  paper:https://arxiv.org/abs/2307.07218 reference: 