Edresson commited on
Commit
0b88aa6
1 Parent(s): b2d1917

Create config_v1.json

Browse files
Files changed (1) hide show
  1. config_v1.json +165 -0
config_v1.json ADDED
@@ -0,0 +1,165 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "output",
3
+ "logger_uri": null,
4
+ "run_name": "run",
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
+ "save_on_interrupt": true,
13
+ "log_model_step": null,
14
+ "save_step": 10000,
15
+ "save_n_checkpoints": 5,
16
+ "save_checkpoints": true,
17
+ "save_all_best": false,
18
+ "save_best_after": 10000,
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": 1000,
29
+ "batch_size": 32,
30
+ "eval_batch_size": 16,
31
+ "grad_clip": 0.0,
32
+ "scheduler_after_epoch": true,
33
+ "lr": 0.001,
34
+ "optimizer": "radam",
35
+ "optimizer_params": null,
36
+ "lr_scheduler": null,
37
+ "lr_scheduler_params": {},
38
+ "use_grad_scaler": false,
39
+ "allow_tf32": false,
40
+ "cudnn_enable": true,
41
+ "cudnn_deterministic": false,
42
+ "cudnn_benchmark": false,
43
+ "training_seed": 54321,
44
+ "model": "xtts",
45
+ "num_loader_workers": 0,
46
+ "num_eval_loader_workers": 0,
47
+ "use_noise_augment": false,
48
+ "audio": {
49
+ "sample_rate": 22050,
50
+ "diffusion_sample_rate": 24000,
51
+ "output_sample_rate": 24000
52
+ },
53
+ "use_phonemes": false,
54
+ "phonemizer": null,
55
+ "phoneme_language": null,
56
+ "compute_input_seq_cache": false,
57
+ "text_cleaner": null,
58
+ "enable_eos_bos_chars": false,
59
+ "test_sentences_file": "",
60
+ "phoneme_cache_path": null,
61
+ "characters": null,
62
+ "add_blank": false,
63
+ "batch_group_size": 0,
64
+ "loss_masking": null,
65
+ "min_audio_len": 1,
66
+ "max_audio_len": Infinity,
67
+ "min_text_len": 1,
68
+ "max_text_len": Infinity,
69
+ "compute_f0": false,
70
+ "compute_energy": false,
71
+ "compute_linear_spec": false,
72
+ "precompute_num_workers": 0,
73
+ "start_by_longest": false,
74
+ "shuffle": false,
75
+ "drop_last": false,
76
+ "datasets": [
77
+ {
78
+ "formatter": "",
79
+ "dataset_name": "",
80
+ "path": "",
81
+ "meta_file_train": "",
82
+ "ignored_speakers": null,
83
+ "language": "",
84
+ "phonemizer": "",
85
+ "meta_file_val": "",
86
+ "meta_file_attn_mask": ""
87
+ }
88
+ ],
89
+ "test_sentences": [],
90
+ "eval_split_max_size": null,
91
+ "eval_split_size": 0.01,
92
+ "use_speaker_weighted_sampler": false,
93
+ "speaker_weighted_sampler_alpha": 1.0,
94
+ "use_language_weighted_sampler": false,
95
+ "language_weighted_sampler_alpha": 1.0,
96
+ "use_length_weighted_sampler": false,
97
+ "length_weighted_sampler_alpha": 1.0,
98
+ "model_args": {
99
+ "gpt_batch_size": 1,
100
+ "enable_redaction": false,
101
+ "use_hifigan": false,
102
+ "kv_cache": true,
103
+ "gpt_checkpoint": null,
104
+ "clvp_checkpoint": null,
105
+ "decoder_checkpoint": null,
106
+ "num_chars": 255,
107
+ "tokenizer_file": "",
108
+ "gpt_max_audio_tokens": 605,
109
+ "gpt_max_text_tokens": 402,
110
+ "gpt_max_prompt_tokens": 70,
111
+ "gpt_layers": 30,
112
+ "gpt_n_model_channels": 1024,
113
+ "gpt_n_heads": 16,
114
+ "gpt_number_text_tokens": 5024,
115
+ "gpt_start_text_token": 261,
116
+ "gpt_stop_text_token": 0,
117
+ "gpt_num_audio_tokens": 8194,
118
+ "gpt_start_audio_token": 8192,
119
+ "gpt_stop_audio_token": 8193,
120
+ "diff_model_channels": 1024,
121
+ "diff_num_layers": 10,
122
+ "diff_in_channels": 100,
123
+ "diff_out_channels": 200,
124
+ "diff_in_latent_channels": 1024,
125
+ "diff_in_tokens": 8193,
126
+ "diff_dropout": 0,
127
+ "diff_use_fp16": false,
128
+ "diff_num_heads": 16,
129
+ "diff_layer_drop": 0,
130
+ "diff_unconditioned_percentage": 0,
131
+ "input_sample_rate": 22050,
132
+ "output_sample_rate": 24000,
133
+ "output_hop_length": 256,
134
+ "ar_mel_length_compression": 1024,
135
+ "decoder_input_dim": 1024,
136
+ "d_vector_dim": 512,
137
+ "cond_d_vector_in_each_upsampling_layer": true,
138
+ "duration_const": 102400
139
+ },
140
+ "model_dir": null,
141
+ "languages": [
142
+ "en",
143
+ "es",
144
+ "fr",
145
+ "de",
146
+ "it",
147
+ "pt",
148
+ "pl",
149
+ "tr",
150
+ "ru",
151
+ "nl",
152
+ "cs",
153
+ "ar",
154
+ "zh-cn"
155
+ ],
156
+ "temperature": 0.2,
157
+ "length_penalty": 1.0,
158
+ "repetition_penalty": 2.0,
159
+ "top_p": 0.8,
160
+ "cond_free_k": 2.0,
161
+ "diffusion_temperature": 1.0,
162
+ "num_gpt_outputs": 16,
163
+ "decoder_iterations": 30,
164
+ "decoder_sampler": "ddim"
165
+ }