Chinese
English
sandy1990418 commited on
Commit
cdfd22b
·
1 Parent(s): 55a36c7

1st version of xtts-v2 with Mandarin accent

Browse files
Files changed (6) hide show
  1. checkpoint.pth +3 -0
  2. config.json +246 -0
  3. dvae.pth +3 -0
  4. mel_stats.pth +3 -0
  5. model.pth +3 -0
  6. vocab.json +0 -0
checkpoint.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36b1d450f803c532393d0aba924fb05f083fa3fbc545235b0ed17d8523d6cd68
3
+ size 5607928213
config.json ADDED
@@ -0,0 +1,246 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "zh-tw-xtts-v2",
3
+ "logger_uri": null,
4
+ "run_name": "GPT_XTTS_v2.0_LJSpeech_FT",
5
+ "project_name": "XTTS_trainer",
6
+ "run_description": "\n GPT XTTS training\n ",
7
+ "print_step": 50,
8
+ "plot_step": 100,
9
+ "model_param_stats": false,
10
+ "wandb_entity": null,
11
+ "dashboard_logger": "tensorboard",
12
+ "save_on_interrupt": true,
13
+ "log_model_step": 100,
14
+ "save_step": 100,
15
+ "save_n_checkpoints": 1,
16
+ "save_checkpoints": true,
17
+ "save_all_best": true,
18
+ "save_best_after": 0,
19
+ "target_loss": null,
20
+ "print_eval": false,
21
+ "test_delay_epochs": 0,
22
+ "run_eval": true,
23
+ "run_eval_steps": null,
24
+ "distributed_backend": "nccl",
25
+ "distributed_url": "tcp://localhost:54321",
26
+ "mixed_precision": false,
27
+ "precision": "fp16",
28
+ "epochs": 200,
29
+ "batch_size": 3,
30
+ "eval_batch_size": 3,
31
+ "grad_clip": 0.0,
32
+ "scheduler_after_epoch": true,
33
+ "lr": 5e-06,
34
+ "optimizer": "AdamW",
35
+ "optimizer_params": {
36
+ "betas": [
37
+ 0.9,
38
+ 0.96
39
+ ],
40
+ "eps": 1e-08,
41
+ "weight_decay": 0.01
42
+ },
43
+ "lr_scheduler": "MultiStepLR",
44
+ "lr_scheduler_params": {
45
+ "milestones": [
46
+ 900000,
47
+ 2700000,
48
+ 5400000
49
+ ],
50
+ "gamma": 0.5,
51
+ "last_epoch": -1
52
+ },
53
+ "use_grad_scaler": false,
54
+ "allow_tf32": false,
55
+ "cudnn_enable": true,
56
+ "cudnn_deterministic": false,
57
+ "cudnn_benchmark": false,
58
+ "training_seed": 1,
59
+ "model": "xtts",
60
+ "num_loader_workers": 8,
61
+ "num_eval_loader_workers": 0,
62
+ "use_noise_augment": false,
63
+ "audio": {
64
+ "sample_rate": 22050,
65
+ "output_sample_rate": 24000,
66
+ "dvae_sample_rate": 22050
67
+ },
68
+ "use_phonemes": false,
69
+ "phonemizer": null,
70
+ "phoneme_language": null,
71
+ "compute_input_seq_cache": false,
72
+ "text_cleaner": null,
73
+ "enable_eos_bos_chars": false,
74
+ "test_sentences_file": "",
75
+ "phoneme_cache_path": null,
76
+ "characters": null,
77
+ "add_blank": false,
78
+ "batch_group_size": 48,
79
+ "loss_masking": null,
80
+ "min_audio_len": 1,
81
+ "max_audio_len": Infinity,
82
+ "min_text_len": 1,
83
+ "max_text_len": Infinity,
84
+ "compute_f0": false,
85
+ "compute_energy": false,
86
+ "compute_linear_spec": false,
87
+ "precompute_num_workers": 0,
88
+ "start_by_longest": false,
89
+ "shuffle": false,
90
+ "drop_last": false,
91
+ "datasets": [
92
+ {
93
+ "formatter": "",
94
+ "dataset_name": "",
95
+ "path": "",
96
+ "meta_file_train": "",
97
+ "ignored_speakers": null,
98
+ "language": "",
99
+ "phonemizer": "",
100
+ "meta_file_val": "",
101
+ "meta_file_attn_mask": ""
102
+ }
103
+ ],
104
+ "test_sentences": [
105
+ {
106
+ "text": "\u8fd8\u662fGOOGLE\u7684ERICSCHMIDT\u4f1a\u8f9e\u53bb\u82f9\u679c\u8463\u4e8b\u4e00\u804c\u5462?",
107
+ "speaker_wav": "../tts_audio/recipes/taiwanvoice/wav/taiwanvoice_1.wav",
108
+ "language": "zh-cn"
109
+ },
110
+ {
111
+ "text": "MLB\u662f\u7f8e\u56fd\u56db\u5927\u804c\u4e1a\u4f53\u80b2\u8054\u76df\u4e4b\u4e00\u3002",
112
+ "speaker_wav": "../tts_audio/recipes/taiwanvoice/wav/taiwanvoice_2.wav",
113
+ "language": "zh-cn"
114
+ },
115
+ {
116
+ "text": "\u8fd9\u9879\u6d3b\u52a8\u7531APC\u7b56\u5212\u3002",
117
+ "speaker_wav": "../tts_audio/recipes/taiwanvoice/wav/taiwanvoice_3.wav",
118
+ "language": "zh-cn"
119
+ },
120
+ {
121
+ "text": "ITS\u5229\u7528\u5148\u8fdb\u4e4b\u7535\u5b50\u3001\u901a\u4fe1\u3001\u7535\u8111\u3001\u63a7\u5236\u53ca\u611f\u6d4b\u7b49\u6280\u672f\u4e8e\u5404\u79cd\u8fd0\u8f93\u7cfb\u7edf\u3002",
122
+ "speaker_wav": "../tts_audio/recipes/taiwanvoice/wav/taiwanvoice_4.wav",
123
+ "language": "zh-cn"
124
+ },
125
+ {
126
+ "text": "NVIDIA\u5bf9GOOGLEANDROID\u5e94\u7528\u5230\u8fd9\u7c7b\u88c5\u7f6e\u6709\u5f88\u9ad8\u7684\u671f\u671b\u3002",
127
+ "speaker_wav": "../tts_audio/recipes/taiwanvoice/wav/taiwanvoice_5.wav",
128
+ "language": "zh-cn"
129
+ },
130
+ {
131
+ "text": "\u53f0\u5927\u533b\u9662\u91d1\u5c71\u5206\u9662",
132
+ "speaker_wav": "../tts_audio/recipes/common_voice/wav/common_voice13_6793.wav",
133
+ "language": "zh-cn"
134
+ },
135
+ {
136
+ "text": "\u53f0\u4e2d\u5e02\u7acb\u56fe\u4e66\u9986\u6c99\u9e7f\u6df1\u6ce2\u5206\u9986",
137
+ "speaker_wav": "../tts_audio/recipes/common_voice/wav/common_voice13_6794.wav",
138
+ "language": "zh-cn"
139
+ },
140
+ {
141
+ "text": "\u65b0\u5e73\u7acb\u4f53\u505c\u8f66\u573a",
142
+ "speaker_wav": "../tts_audio/recipes/common_voice/wav/common_voice13_6795.wav",
143
+ "language": "zh-cn"
144
+ },
145
+ {
146
+ "text": "\u516d\u987a\u591a\u529f\u80fd\u6d3b\u52a8\u4e2d\u5fc3",
147
+ "speaker_wav": "../tts_audio/recipes/common_voice/wav/common_voice13_6796.wav",
148
+ "language": "zh-cn"
149
+ },
150
+ {
151
+ "text": "\u4eba\u56e0\u5e7b\u60f3\u800c\u4f1f\u5927",
152
+ "speaker_wav": "../tts_audio/recipes/common_voice/wav/common_voice13_6797.wav",
153
+ "language": "zh-cn"
154
+ },
155
+ {
156
+ "text": "\u5982\u679c\u4f60\u4e5f\u6709\u5174\u8da3\uff0c\u6b22\u8fce\u4e00\u540c\u9009\u8d2d",
157
+ "speaker_wav": "../tts_audio/recipes/common_voice/wav/common_voice13_6798.wav",
158
+ "language": "zh-cn"
159
+ }
160
+ ],
161
+ "eval_split_max_size": 256,
162
+ "eval_split_size": 0.01,
163
+ "use_speaker_weighted_sampler": false,
164
+ "speaker_weighted_sampler_alpha": 1.0,
165
+ "use_language_weighted_sampler": false,
166
+ "language_weighted_sampler_alpha": 1.0,
167
+ "use_length_weighted_sampler": false,
168
+ "length_weighted_sampler_alpha": 1.0,
169
+ "model_args": {
170
+ "gpt_batch_size": 1,
171
+ "enable_redaction": false,
172
+ "kv_cache": true,
173
+ "gpt_checkpoint": "",
174
+ "clvp_checkpoint": null,
175
+ "decoder_checkpoint": null,
176
+ "num_chars": 255,
177
+ "tokenizer_file": "zh-tw-xtts-v2/vocab.json",
178
+ "gpt_max_audio_tokens": 605,
179
+ "gpt_max_text_tokens": 402,
180
+ "gpt_max_prompt_tokens": 70,
181
+ "gpt_layers": 30,
182
+ "gpt_n_model_channels": 1024,
183
+ "gpt_n_heads": 16,
184
+ "gpt_number_text_tokens": 6681,
185
+ "gpt_start_text_token": 261,
186
+ "gpt_stop_text_token": 0,
187
+ "gpt_num_audio_tokens": 1026,
188
+ "gpt_start_audio_token": 1024,
189
+ "gpt_stop_audio_token": 1025,
190
+ "gpt_code_stride_len": 1024,
191
+ "gpt_use_masking_gt_prompt_approach": true,
192
+ "gpt_use_perceiver_resampler": true,
193
+ "input_sample_rate": 22050,
194
+ "output_sample_rate": 24000,
195
+ "output_hop_length": 256,
196
+ "decoder_input_dim": 1024,
197
+ "d_vector_dim": 512,
198
+ "cond_d_vector_in_each_upsampling_layer": true,
199
+ "duration_const": 102400,
200
+ "min_conditioning_length": 66150,
201
+ "max_conditioning_length": 132300,
202
+ "gpt_loss_text_ce_weight": 0.1,
203
+ "gpt_loss_mel_ce_weight": 0.9,
204
+ "debug_loading_failures": false,
205
+ "max_wav_length": 255995,
206
+ "max_text_length": 200,
207
+ "mel_norm_file": "zh-tw-xtts-v2/mel_stats.pth",
208
+ "dvae_checkpoint": "zh-tw-xtts-v2/dvae.pth",
209
+ "xtts_checkpoint": "zh-tw-xtts-v2/model.pth",
210
+ "vocoder": ""
211
+ },
212
+ "model_dir": null,
213
+ "languages": [
214
+ "en",
215
+ "es",
216
+ "fr",
217
+ "de",
218
+ "it",
219
+ "pt",
220
+ "pl",
221
+ "tr",
222
+ "ru",
223
+ "nl",
224
+ "cs",
225
+ "ar",
226
+ "zh-cn",
227
+ "hu",
228
+ "ko",
229
+ "ja",
230
+ "hi"
231
+ ],
232
+ "temperature": 0.85,
233
+ "length_penalty": 1.0,
234
+ "repetition_penalty": 2.0,
235
+ "top_k": 50,
236
+ "top_p": 0.85,
237
+ "num_gpt_outputs": 1,
238
+ "gpt_cond_len": 12,
239
+ "gpt_cond_chunk_len": 4,
240
+ "max_ref_len": 10,
241
+ "sound_norm_refs": false,
242
+ "optimizer_wd_only_on_weights": true,
243
+ "weighted_loss_attrs": {},
244
+ "weighted_loss_multipliers": {},
245
+ "github_branch": "* dev"
246
+ }
dvae.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b29bc227d410d4991e0a8c09b858f77415013eeb9fba9650258e96095557d97a
3
+ size 210514388
mel_stats.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f69422a8a8f344c4fca2f0c6b8d41d2151d6615b7321e48e6bb15ae949b119c
3
+ size 1067
model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7ea20001c6a0a841c77e252d8409f6a74fb423e79b3206a0771ba5989776187
3
+ size 1867929118
vocab.json ADDED
The diff for this file is too large to render. See raw diff