├── .gitignore ├── LICENSE ├── README.md ├── README_ZH.md ├── assets ├── docs │ ├── voice_clone.md │ └── voice_clone_ZH.md └── speakers │ └── 2222.pt ├── chattts_plus ├── __init__.py ├── commons │ ├── __init__.py │ ├── constants.py │ ├── logger.py │ ├── norm.py │ ├── onnx2trt.py │ ├── text_utils.py │ └── utils.py ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ └── collator.py ├── models │ ├── __init__.py │ ├── dvae.py │ ├── gpt.py │ ├── llama.py │ ├── processors.py │ └── tokenizer.py ├── pipelines │ ├── __init__.py │ └── chattts_plus_pipeline.py └── trt_models │ ├── __init__.py │ ├── base_model.py │ ├── gpt_trt.py │ ├── llama_trt_model.py │ └── predictor.py ├── configs ├── accelerate │ └── deepspeed_config.yaml ├── infer │ ├── chattts_plus.yaml │ └── chattts_plus_trt.yaml └── train │ ├── train_speaker_embedding.yaml │ └── train_voice_clone_lora.yaml ├── demos └── notebooklm-podcast │ ├── extract_files_to_texts.py │ ├── llm_api.py │ ├── requirements.txt │ ├── speaker_pt │ ├── en_man_5200.pt │ ├── en_man_8200.pt │ ├── en_man_9400.pt │ ├── en_man_9500.pt │ ├── en_woman_1200.pt │ ├── en_woman_4600.pt │ ├── en_woman_5600.pt │ ├── zh_man_1888.pt │ ├── zh_man_2155.pt │ ├── zh_man_54.pt │ ├── zh_woman_1528.pt │ ├── zh_woman_492.pt │ └── zh_woman_621.pt │ ├── tts.py │ └── utils.py ├── requirements.txt ├── scripts └── conversions │ └── convert_train_list.py ├── setup.py ├── tests ├── test_models.py └── test_pipelines.py ├── train_lora.py ├── update.bat ├── webui.bat └── webui.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/README.md -------------------------------------------------------------------------------- /README_ZH.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/README_ZH.md -------------------------------------------------------------------------------- /assets/docs/voice_clone.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/assets/docs/voice_clone.md -------------------------------------------------------------------------------- /assets/docs/voice_clone_ZH.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/assets/docs/voice_clone_ZH.md -------------------------------------------------------------------------------- /assets/speakers/2222.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/assets/speakers/2222.pt -------------------------------------------------------------------------------- /chattts_plus/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/__init__.py -------------------------------------------------------------------------------- /chattts_plus/commons/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/commons/__init__.py -------------------------------------------------------------------------------- /chattts_plus/commons/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/commons/constants.py -------------------------------------------------------------------------------- /chattts_plus/commons/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/commons/logger.py -------------------------------------------------------------------------------- /chattts_plus/commons/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/commons/norm.py -------------------------------------------------------------------------------- /chattts_plus/commons/onnx2trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/commons/onnx2trt.py -------------------------------------------------------------------------------- /chattts_plus/commons/text_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/commons/text_utils.py -------------------------------------------------------------------------------- /chattts_plus/commons/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/commons/utils.py -------------------------------------------------------------------------------- /chattts_plus/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/datasets/__init__.py -------------------------------------------------------------------------------- /chattts_plus/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/datasets/base_dataset.py -------------------------------------------------------------------------------- /chattts_plus/datasets/collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/datasets/collator.py -------------------------------------------------------------------------------- /chattts_plus/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/models/__init__.py -------------------------------------------------------------------------------- /chattts_plus/models/dvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/models/dvae.py -------------------------------------------------------------------------------- /chattts_plus/models/gpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/models/gpt.py -------------------------------------------------------------------------------- /chattts_plus/models/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/models/llama.py -------------------------------------------------------------------------------- /chattts_plus/models/processors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/models/processors.py -------------------------------------------------------------------------------- /chattts_plus/models/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/models/tokenizer.py -------------------------------------------------------------------------------- /chattts_plus/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/pipelines/__init__.py -------------------------------------------------------------------------------- /chattts_plus/pipelines/chattts_plus_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/pipelines/chattts_plus_pipeline.py -------------------------------------------------------------------------------- /chattts_plus/trt_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/trt_models/__init__.py -------------------------------------------------------------------------------- /chattts_plus/trt_models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/trt_models/base_model.py -------------------------------------------------------------------------------- /chattts_plus/trt_models/gpt_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/trt_models/gpt_trt.py -------------------------------------------------------------------------------- /chattts_plus/trt_models/llama_trt_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/trt_models/llama_trt_model.py -------------------------------------------------------------------------------- /chattts_plus/trt_models/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/chattts_plus/trt_models/predictor.py -------------------------------------------------------------------------------- /configs/accelerate/deepspeed_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/configs/accelerate/deepspeed_config.yaml -------------------------------------------------------------------------------- /configs/infer/chattts_plus.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/configs/infer/chattts_plus.yaml -------------------------------------------------------------------------------- /configs/infer/chattts_plus_trt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/configs/infer/chattts_plus_trt.yaml -------------------------------------------------------------------------------- /configs/train/train_speaker_embedding.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/configs/train/train_speaker_embedding.yaml -------------------------------------------------------------------------------- /configs/train/train_voice_clone_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/configs/train/train_voice_clone_lora.yaml -------------------------------------------------------------------------------- /demos/notebooklm-podcast/extract_files_to_texts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/extract_files_to_texts.py -------------------------------------------------------------------------------- /demos/notebooklm-podcast/llm_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/llm_api.py -------------------------------------------------------------------------------- /demos/notebooklm-podcast/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/requirements.txt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/en_man_5200.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/en_man_5200.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/en_man_8200.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/en_man_8200.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/en_man_9400.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/en_man_9400.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/en_man_9500.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/en_man_9500.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/en_woman_1200.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/en_woman_1200.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/en_woman_4600.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/en_woman_4600.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/en_woman_5600.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/en_woman_5600.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/zh_man_1888.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/zh_man_1888.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/zh_man_2155.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/zh_man_2155.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/zh_man_54.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/zh_man_54.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/zh_woman_1528.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/zh_woman_1528.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/zh_woman_492.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/zh_woman_492.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/speaker_pt/zh_woman_621.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/speaker_pt/zh_woman_621.pt -------------------------------------------------------------------------------- /demos/notebooklm-podcast/tts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/tts.py -------------------------------------------------------------------------------- /demos/notebooklm-podcast/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/demos/notebooklm-podcast/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/conversions/convert_train_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/scripts/conversions/convert_train_list.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/test_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/tests/test_pipelines.py -------------------------------------------------------------------------------- /train_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/train_lora.py -------------------------------------------------------------------------------- /update.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/update.bat -------------------------------------------------------------------------------- /webui.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/webui.bat -------------------------------------------------------------------------------- /webui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/warmshao/ChatTTSPlus/HEAD/webui.py --------------------------------------------------------------------------------