Text-to-Speech
TTS
guymandude commited on
Commit
5a8b966
1 Parent(s): 0a7eb70

Upload 4 files

Browse files
Files changed (4) hide show
  1. config.json +558 -0
  2. language_ids.json +13 -0
  3. model_speakers.pth +3 -0
  4. vits_11_ZA_model.pth +3 -0
config.json ADDED
@@ -0,0 +1,558 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "./models/11-ZA_multilingual",
3
+ "logger_uri": null,
4
+ "run_name": "11_ZA",
5
+ "project_name": null,
6
+ "run_description": "\ud83d\udc38Coqui trainer run.",
7
+ "print_step": 25,
8
+ "plot_step": 100,
9
+ "model_param_stats": false,
10
+ "wandb_entity": null,
11
+ "dashboard_logger": "tensorboard",
12
+ "log_model_step": 5000,
13
+ "save_step": 5000,
14
+ "save_n_checkpoints": 5,
15
+ "save_checkpoints": true,
16
+ "save_all_best": false,
17
+ "save_best_after": 10000,
18
+ "target_loss": null,
19
+ "print_eval": false,
20
+ "test_delay_epochs": -1,
21
+ "run_eval": true,
22
+ "run_eval_steps": null,
23
+ "distributed_backend": "nccl",
24
+ "distributed_url": "tcp://localhost:54321",
25
+ "mixed_precision": true,
26
+ "epochs": 1000,
27
+ "batch_size": 16,
28
+ "eval_batch_size": 16,
29
+ "grad_clip": [
30
+ 1000,
31
+ 1000
32
+ ],
33
+ "scheduler_after_epoch": true,
34
+ "lr": 0.001,
35
+ "optimizer": "AdamW",
36
+ "optimizer_params": {
37
+ "betas": [
38
+ 0.8,
39
+ 0.99
40
+ ],
41
+ "eps": 1e-09,
42
+ "weight_decay": 0.01
43
+ },
44
+ "lr_scheduler": null,
45
+ "lr_scheduler_params": {},
46
+ "use_grad_scaler": false,
47
+ "cudnn_enable": true,
48
+ "cudnn_deterministic": false,
49
+ "cudnn_benchmark": false,
50
+ "training_seed": 54321,
51
+ "model": "vits",
52
+ "num_loader_workers": 4,
53
+ "num_eval_loader_workers": 4,
54
+ "use_noise_augment": false,
55
+ "audio": {
56
+ "fft_size": 1024,
57
+ "sample_rate": 16000,
58
+ "win_length": 1024,
59
+ "hop_length": 256,
60
+ "num_mels": 80,
61
+ "mel_fmin": 0,
62
+ "mel_fmax": null
63
+ },
64
+ "use_phonemes": false,
65
+ "phonemizer": null,
66
+ "phoneme_language": null,
67
+ "compute_input_seq_cache": true,
68
+ "text_cleaner": "multilingual_cleaners",
69
+ "enable_eos_bos_chars": false,
70
+ "test_sentences_file": "",
71
+ "phoneme_cache_path": null,
72
+ "characters": {
73
+ "characters_class": "TTS.tts.models.vits.VitsCharacters",
74
+ "vocab_dict": null,
75
+ "pad": "<PAD>",
76
+ "eos": "<EOS>",
77
+ "bos": "<BOS>",
78
+ "blank": "<BLNK>",
79
+ "characters": "\u014b\u0266\u0254ma\u02d0spfut\u031dkr\u026ad\u025b!\u00a1'(),\u2014-.:;\u00bf?abcdefghijklmnopqrstuvwxyz\u00b5\u00df\u00e0\u00e1\u00e2\u00e4\u00e5\u00e6\u00e7\u00e8\u00e9\u00ea\u00eb\u00ec\u00ed\u00ee\u00ef\u00f1\u00f2\u00f3\u00f4\u00f6\u00f9\u00fa\u00fb\u00fc\u0105\u0107\u0119\u0142\u0144\u0153\u015b\u015f\u017a\u017c\u0192\u0430\u0431\u0432\u0433\u0434\u0435\u0436\u0437\u0438\u0439\u043a\u043b\u043c\u043d\u043e\u043f\u0440\u0441\u0442\u0443\u0444\u0445\u0446\u0447\u0448\u0449\u044a\u044b\u044c\u044d\u044e\u044f\u0451\u0454\u0456\u0457\u0491\u04e7 \u00ab\u00b0\u00b1\u00b5\u00bb$%&\u2018\u2019\u201a\u201c`\u201d\u201e",
80
+ "punctuations": "!\u00a1'(),-.:;\u00bf? ",
81
+ "phonemes": "\u014b\u0266\u0254ma\u02d0spfut\u031dkr\u026ad\u025baa\u02d0a\u0303\u02d0bcdefii\u02d0jklmnoprstuu\u02d0vwxyy\u02d0\u00e6\u02d0\u00f8\u014b\u0153\u0153\u02d0\u0254\u0254\u02d0\u0254\u0303\u02d0\u0258\u025b\u025b\u02d0\u025b\u0303\u02d0\u0261\u0266\u026a\u031d",
82
+ "is_unique": true,
83
+ "is_sorted": true
84
+ },
85
+ "add_blank": false,
86
+ "batch_group_size": 0,
87
+ "loss_masking": null,
88
+ "min_audio_len": 32768,
89
+ "max_audio_len": 160000,
90
+ "min_text_len": 1,
91
+ "max_text_len": Infinity,
92
+ "compute_f0": false,
93
+ "compute_energy": false,
94
+ "compute_linear_spec": true,
95
+ "precompute_num_workers": 0,
96
+ "start_by_longest": false,
97
+ "shuffle": false,
98
+ "drop_last": false,
99
+ "datasets": [
100
+ {
101
+ "formatter": "nchlt",
102
+ "dataset_name": "tts_eng",
103
+ "path": "",
104
+ "meta_file_train": "eng_metadata.txt",
105
+ "ignored_speakers": null,
106
+ "language": "eng",
107
+ "phonemizer": "",
108
+ "meta_file_val": "",
109
+ "meta_file_attn_mask": ""
110
+ },
111
+ {
112
+ "formatter": "nchlt",
113
+ "dataset_name": "tts_afr",
114
+ "path": "",
115
+ "meta_file_train": "afr_metadata.txt",
116
+ "ignored_speakers": null,
117
+ "language": "afr",
118
+ "phonemizer": "",
119
+ "meta_file_val": "",
120
+ "meta_file_attn_mask": ""
121
+ },
122
+ {
123
+ "formatter": "nchlt",
124
+ "dataset_name": "tts_zul",
125
+ "path": "",
126
+ "meta_file_train": "zul_metadata.txt",
127
+ "ignored_speakers": null,
128
+ "language": "zul",
129
+ "phonemizer": "",
130
+ "meta_file_val": "",
131
+ "meta_file_attn_mask": ""
132
+ },
133
+ {
134
+ "formatter": "nchlt",
135
+ "dataset_name": "tts_zho",
136
+ "path": "",
137
+ "meta_file_train": "xho_metadata.txt",
138
+ "ignored_speakers": null,
139
+ "language": "xho",
140
+ "phonemizer": "",
141
+ "meta_file_val": "",
142
+ "meta_file_attn_mask": ""
143
+ },
144
+ {
145
+ "formatter": "nchlt",
146
+ "dataset_name": "tts_ven",
147
+ "path": "",
148
+ "meta_file_train": "ven_metadata.txt",
149
+ "ignored_speakers": null,
150
+ "language": "ven",
151
+ "phonemizer": "",
152
+ "meta_file_val": "",
153
+ "meta_file_attn_mask": ""
154
+ },
155
+ {
156
+ "formatter": "nchlt",
157
+ "dataset_name": "tts_tso",
158
+ "path": "",
159
+ "meta_file_train": "tso_metadata.txt",
160
+ "ignored_speakers": null,
161
+ "language": "tso",
162
+ "phonemizer": "",
163
+ "meta_file_val": "",
164
+ "meta_file_attn_mask": ""
165
+ },
166
+ {
167
+ "formatter": "nchlt",
168
+ "dataset_name": "tts_tsn",
169
+ "path": "",
170
+ "meta_file_train": "tsn_metadata.txt",
171
+ "ignored_speakers": null,
172
+ "language": "tsn",
173
+ "phonemizer": "",
174
+ "meta_file_val": "",
175
+ "meta_file_attn_mask": ""
176
+ },
177
+ {
178
+ "formatter": "nchlt",
179
+ "dataset_name": "tts_sot",
180
+ "path": "",
181
+ "meta_file_train": "sot_metadata.txt",
182
+ "ignored_speakers": null,
183
+ "language": "sot",
184
+ "phonemizer": "",
185
+ "meta_file_val": "",
186
+ "meta_file_attn_mask": ""
187
+ },
188
+ {
189
+ "formatter": "nchlt",
190
+ "dataset_name": "tts_ssw",
191
+ "path": "",
192
+ "meta_file_train": "ssw_metadata.txt",
193
+ "ignored_speakers": null,
194
+ "language": "ssw",
195
+ "phonemizer": "",
196
+ "meta_file_val": "",
197
+ "meta_file_attn_mask": ""
198
+ },
199
+ {
200
+ "formatter": "nchlt",
201
+ "dataset_name": "tts_nbl",
202
+ "path": "",
203
+ "meta_file_train": "nbl_metadata.txt",
204
+ "ignored_speakers": null,
205
+ "language": "nbl",
206
+ "phonemizer": "",
207
+ "meta_file_val": "",
208
+ "meta_file_attn_mask": ""
209
+ },
210
+ {
211
+ "formatter": "nchlt",
212
+ "dataset_name": "tts_nso",
213
+ "path": "",
214
+ "meta_file_train": "nso_metadata.txt",
215
+ "ignored_speakers": null,
216
+ "language": "nso",
217
+ "phonemizer": "",
218
+ "meta_file_val": "",
219
+ "meta_file_attn_mask": ""
220
+ }
221
+ ],
222
+ "test_sentences": [
223
+ [
224
+ "Something has to be put in place.",
225
+ "p244",
226
+ null,
227
+ "eng"
228
+ ],
229
+ [
230
+ "The gaseous matters (carbon dioxide, oxygen, nitrogen) enter the plant mainly by diffusion from the atmosphere,",
231
+ "lq_eng_f",
232
+ null,
233
+ "eng"
234
+ ],
235
+ [
236
+ "Anyone who was familiar with that area of Dallas would have known that the motorcade would probably pass the Texas School Book Depository to get from Main Street",
237
+ "Leah",
238
+ null,
239
+ "eng"
240
+ ],
241
+ [
242
+ "verbruik gedurende dieselfde",
243
+ "lq_afr_f",
244
+ null,
245
+ "afr"
246
+ ],
247
+ [
248
+ "In ons howe wil ons 'n regering van wette h\u00ea en nie van mans nie.",
249
+ "Willem",
250
+ null,
251
+ "afr"
252
+ ],
253
+ [
254
+ "persvrystellings aanbiedings tentoon",
255
+ "adri",
256
+ null,
257
+ "afr"
258
+ ],
259
+ [
260
+ "siphunyuke olakeni lukamama",
261
+ "lq_zul_f",
262
+ null,
263
+ "zul"
264
+ ],
265
+ [
266
+ "Abampofu bezwe elizungezile bahlala izigodlo zalo ezihlakazekile.",
267
+ "thando",
268
+ null,
269
+ "zul"
270
+ ],
271
+ [
272
+ "Futhi eyodwa enikeze izaba zokuphikisana nephakheji elinobuningi ekuseni ngakusasa ekuseni.",
273
+ "Themba",
274
+ null,
275
+ "zul"
276
+ ],
277
+ [
278
+ "le tjhabile le monatjana ho",
279
+ "lq_sot_f",
280
+ null,
281
+ "sot"
282
+ ],
283
+ [
284
+ "ha o ne o mpolella",
285
+ "lq_sot_f",
286
+ null,
287
+ "sot"
288
+ ],
289
+ [
290
+ "Ho sheba eufrate e kholo e koahelang phuleng ka tlase.",
291
+ "lq_sot_f",
292
+ null,
293
+ "sot"
294
+ ],
295
+ [
296
+ "pheji leyi ku fanele ku",
297
+ "lq_tso_f",
298
+ null,
299
+ "tso"
300
+ ],
301
+ [
302
+ "kota ku hlayela no langutisela",
303
+ "lq_tso_f",
304
+ null,
305
+ "tso"
306
+ ],
307
+ [
308
+ "vanwa milenge va kotaku tsutsuma",
309
+ "lq_tso_f",
310
+ null,
311
+ "tso"
312
+ ],
313
+ [
314
+ "motsu o o mo letshwaong",
315
+ "lq_tsn_f",
316
+ null,
317
+ "tsn"
318
+ ],
319
+ [
320
+ "itshokodisetsa eng tota ausi thando",
321
+ "lq_tsn_f",
322
+ null,
323
+ "tsn"
324
+ ],
325
+ [
326
+ "la kwa bogodimong jwa bolaodi",
327
+ "lq_tsn_f",
328
+ null,
329
+ "tsn"
330
+ ],
331
+ [
332
+ "kesinye nesinye isikhathi",
333
+ "lq_nbl_m",
334
+ null,
335
+ "nbl"
336
+ ],
337
+ [
338
+ "kibo boke abatlolisiweko",
339
+ "lq_nbl_m",
340
+ null,
341
+ "nbl"
342
+ ],
343
+ [
344
+ "kwamalungelo wabantu kanye",
345
+ "lq_nbl_m",
346
+ null,
347
+ "nbl"
348
+ ],
349
+ [
350
+ "nga thusa vhathu kana vhafarisi",
351
+ "lq_ven_m",
352
+ null,
353
+ "ven"
354
+ ],
355
+ [
356
+ "shumiselwaho vhutali i do tavhanyisa",
357
+ "lq_ven_m",
358
+ null,
359
+ "ven"
360
+ ],
361
+ [
362
+ "badela mushumi we a lova",
363
+ "lq_ven_m",
364
+ null,
365
+ "ven"
366
+ ],
367
+ [
368
+ "nanobe kuneluhlelo kwakhiwe",
369
+ "lq_ssw_f",
370
+ null,
371
+ "ssw"
372
+ ],
373
+ [
374
+ "umcimbi wekugidza umnandzi",
375
+ "lq_ssw_f",
376
+ null,
377
+ "ssw"
378
+ ],
379
+ [
380
+ "lomunye sitijabulisa ngawo",
381
+ "lq_ssw_f",
382
+ null,
383
+ "ssw"
384
+ ],
385
+ [
386
+ "ga setso se se itsego",
387
+ "lq_nso_m",
388
+ null,
389
+ "nso"
390
+ ],
391
+ [
392
+ "motho a ka ba a",
393
+ "lq_nso_m",
394
+ null,
395
+ "nso"
396
+ ],
397
+ [
398
+ "O ile a bolela gape gore o tla diri\u0161a sethunya ge e ba sebaka se ka t\u0161welela.",
399
+ "lq_nso_m",
400
+ null,
401
+ "nso"
402
+ ],
403
+ [
404
+ "ezigqibeleleyo zezemidlalo nezokuziphumza",
405
+ "lq_xho_m",
406
+ null,
407
+ "xho"
408
+ ],
409
+ [
410
+ "Ngeendlela zokufumana impilo ngokunyaniseka ukuba ilahlekile.",
411
+ "lq_xho_m",
412
+ null,
413
+ "xho"
414
+ ],
415
+ [
416
+ "nemeko yokuqesha yahlukene",
417
+ "lq_xho_f",
418
+ null,
419
+ "xho"
420
+ ]
421
+ ],
422
+ "eval_split_max_size": null,
423
+ "eval_split_size": 0.01,
424
+ "use_speaker_weighted_sampler": true,
425
+ "speaker_weighted_sampler_alpha": 1.0,
426
+ "use_language_weighted_sampler": true,
427
+ "language_weighted_sampler_alpha": 1.0,
428
+ "use_length_weighted_sampler": false,
429
+ "length_weighted_sampler_alpha": 1.0,
430
+ "model_args": {
431
+ "num_chars": 236,
432
+ "out_channels": 513,
433
+ "spec_segment_size": 32,
434
+ "hidden_channels": 192,
435
+ "hidden_channels_ffn_text_encoder": 768,
436
+ "num_heads_text_encoder": 2,
437
+ "num_layers_text_encoder": 6,
438
+ "kernel_size_text_encoder": 3,
439
+ "dropout_p_text_encoder": 0.1,
440
+ "dropout_p_duration_predictor": 0.5,
441
+ "kernel_size_posterior_encoder": 5,
442
+ "dilation_rate_posterior_encoder": 1,
443
+ "num_layers_posterior_encoder": 16,
444
+ "kernel_size_flow": 5,
445
+ "dilation_rate_flow": 1,
446
+ "num_layers_flow": 4,
447
+ "resblock_type_decoder": "1",
448
+ "resblock_kernel_sizes_decoder": [
449
+ 3,
450
+ 7,
451
+ 11
452
+ ],
453
+ "resblock_dilation_sizes_decoder": [
454
+ [
455
+ 1,
456
+ 3,
457
+ 5
458
+ ],
459
+ [
460
+ 1,
461
+ 3,
462
+ 5
463
+ ],
464
+ [
465
+ 1,
466
+ 3,
467
+ 5
468
+ ]
469
+ ],
470
+ "upsample_rates_decoder": [
471
+ 8,
472
+ 8,
473
+ 2,
474
+ 2
475
+ ],
476
+ "upsample_initial_channel_decoder": 512,
477
+ "upsample_kernel_sizes_decoder": [
478
+ 16,
479
+ 16,
480
+ 4,
481
+ 4
482
+ ],
483
+ "periods_multi_period_discriminator": [
484
+ 2,
485
+ 3,
486
+ 5,
487
+ 7,
488
+ 11
489
+ ],
490
+ "use_sdp": false,
491
+ "noise_scale": 1.0,
492
+ "inference_noise_scale": 0.667,
493
+ "length_scale": 1,
494
+ "noise_scale_dp": 1.0,
495
+ "inference_noise_scale_dp": 1.0,
496
+ "max_inference_len": null,
497
+ "init_discriminator": false,
498
+ "use_spectral_norm_disriminator": false,
499
+ "use_speaker_embedding": true,
500
+ "num_speakers": 130,
501
+ "speakers_file": "./models/11-ZA_multilingual/11_ZA-May-16-2023_12+09PM-0000000/speakers.pth",
502
+ "d_vector_file": null,
503
+ "speaker_embedding_channels": 256,
504
+ "use_d_vector_file": false,
505
+ "d_vector_dim": 0,
506
+ "detach_dp_input": true,
507
+ "use_language_embedding": true,
508
+ "embedded_language_dim": 12,
509
+ "num_languages": 11,
510
+ "language_ids_file": "./models/11-ZA_multilingual/11_ZA-May-16-2023_12+09PM-0000000/language_ids.json",
511
+ "use_speaker_encoder_as_loss": false,
512
+ "speaker_encoder_config_path": "",
513
+ "speaker_encoder_model_path": "",
514
+ "condition_dp_on_speaker": true,
515
+ "freeze_encoder": false,
516
+ "freeze_DP": false,
517
+ "freeze_PE": false,
518
+ "freeze_flow_decoder": false,
519
+ "freeze_waveform_decoder": false,
520
+ "encoder_sample_rate": null,
521
+ "interpolate_z": true,
522
+ "reinit_DP": false,
523
+ "reinit_text_encoder": false
524
+ },
525
+ "lr_gen": 0.0002,
526
+ "lr_disc": 0.0002,
527
+ "lr_scheduler_gen": "ExponentialLR",
528
+ "lr_scheduler_gen_params": {
529
+ "gamma": 0.999875,
530
+ "last_epoch": -1
531
+ },
532
+ "lr_scheduler_disc": "ExponentialLR",
533
+ "lr_scheduler_disc_params": {
534
+ "gamma": 0.999875,
535
+ "last_epoch": -1
536
+ },
537
+ "kl_loss_alpha": 1.0,
538
+ "disc_loss_alpha": 1.0,
539
+ "gen_loss_alpha": 1.0,
540
+ "feat_loss_alpha": 1.0,
541
+ "mel_loss_alpha": 45.0,
542
+ "dur_loss_alpha": 1.0,
543
+ "speaker_encoder_loss_alpha": 1.0,
544
+ "return_wav": true,
545
+ "use_weighted_sampler": false,
546
+ "weighted_sampler_attrs": {},
547
+ "weighted_sampler_multipliers": {},
548
+ "r": 1,
549
+ "num_speakers": 0,
550
+ "use_speaker_embedding": true,
551
+ "speakers_file": "./models/11-ZA_multilingual/11_ZA-May-16-2023_12+09PM-0000000/speakers.pth",
552
+ "speaker_embedding_channels": 256,
553
+ "language_ids_file": "./models/11-ZA_multilingual/11_ZA-May-16-2023_12+09PM-0000000/language_ids.json",
554
+ "use_language_embedding": true,
555
+ "use_d_vector_file": false,
556
+ "d_vector_file": null,
557
+ "d_vector_dim": 0
558
+ }
language_ids.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "afr": 0,
3
+ "eng": 1,
4
+ "nbl": 2,
5
+ "nso": 3,
6
+ "sot": 4,
7
+ "ssw": 5,
8
+ "tsn": 6,
9
+ "tso": 7,
10
+ "ven": 8,
11
+ "xho": 9,
12
+ "zul": 10
13
+ }
model_speakers.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6f769b5fa40838d291024d57363d0abac06cae651f9bdc4da16a853435f7731
3
+ size 2941
vits_11_ZA_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d49bf95b0e87742db43b77b1c16ce9e9266dd6731ab7c30c7b9bf3129747bd96
3
+ size 157017425