├── .gitignore ├── LICENSE ├── README.md ├── cli.py ├── config └── config_gan.yaml ├── configs ├── config_adm.yaml ├── config_gan.yaml └── config_plm.yaml ├── examples └── mel_step_400k_re_loss_0.4771.png ├── infer.py ├── models ├── megatts2.py └── trainer.py ├── modules ├── __init__.py ├── convnet.py ├── datamodule.py ├── dscrm.py ├── embedding.py ├── mrte.py ├── quantization │ ├── __init__.py │ ├── ac.py │ ├── core_vq.py │ └── vq.py ├── tokenizer.py ├── transformer.py └── vqpe.py ├── prepare_ds.py ├── requirements.txt └── utils ├── distrib.py ├── mandarin_pinyin_to_mfa_lty.dict ├── symbol_table.py ├── textgrid.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/README.md -------------------------------------------------------------------------------- /cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/cli.py -------------------------------------------------------------------------------- /config/config_gan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/config/config_gan.yaml -------------------------------------------------------------------------------- /configs/config_adm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/configs/config_adm.yaml -------------------------------------------------------------------------------- /configs/config_gan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/configs/config_gan.yaml -------------------------------------------------------------------------------- /configs/config_plm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/configs/config_plm.yaml -------------------------------------------------------------------------------- /examples/mel_step_400k_re_loss_0.4771.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/examples/mel_step_400k_re_loss_0.4771.png -------------------------------------------------------------------------------- /infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/infer.py -------------------------------------------------------------------------------- /models/megatts2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/models/megatts2.py -------------------------------------------------------------------------------- /models/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/models/trainer.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/convnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/convnet.py -------------------------------------------------------------------------------- /modules/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/datamodule.py -------------------------------------------------------------------------------- /modules/dscrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/dscrm.py -------------------------------------------------------------------------------- /modules/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/embedding.py -------------------------------------------------------------------------------- /modules/mrte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/mrte.py -------------------------------------------------------------------------------- /modules/quantization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/quantization/__init__.py -------------------------------------------------------------------------------- /modules/quantization/ac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/quantization/ac.py -------------------------------------------------------------------------------- /modules/quantization/core_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/quantization/core_vq.py -------------------------------------------------------------------------------- /modules/quantization/vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/quantization/vq.py -------------------------------------------------------------------------------- /modules/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/tokenizer.py -------------------------------------------------------------------------------- /modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/transformer.py -------------------------------------------------------------------------------- /modules/vqpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/modules/vqpe.py -------------------------------------------------------------------------------- /prepare_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/prepare_ds.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/distrib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/utils/distrib.py -------------------------------------------------------------------------------- /utils/mandarin_pinyin_to_mfa_lty.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/utils/mandarin_pinyin_to_mfa_lty.dict -------------------------------------------------------------------------------- /utils/symbol_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/utils/symbol_table.py -------------------------------------------------------------------------------- /utils/textgrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/utils/textgrid.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LSimon95/megatts2/HEAD/utils/utils.py --------------------------------------------------------------------------------