| { | |
| "output_path": "/root/TTS-XTTS-decoder/checkpoints_xtts", | |
| "logger_uri": null, | |
| "run_name": "GPT_XTTS_v2.0_pt", | |
| "project_name": "XTTS_trainer", | |
| "run_description": "\n GPT XTTS training\n ", | |
| "print_step": 50, | |
| "plot_step": 100, | |
| "model_param_stats": false, | |
| "wandb_entity": null, | |
| "dashboard_logger": "tensorboard", | |
| "save_on_interrupt": true, | |
| "log_model_step": 1000, | |
| "save_step": 10000, | |
| "save_n_checkpoints": 3, | |
| "save_checkpoints": true, | |
| "save_all_best": false, | |
| "save_best_after": 0, | |
| "target_loss": null, | |
| "print_eval": true, | |
| "test_delay_epochs": 0, | |
| "run_eval": true, | |
| "run_eval_steps": 10000, | |
| "distributed_backend": "nccl", | |
| "distributed_url": "tcp://localhost:54321", | |
| "mixed_precision": false, | |
| "precision": "fp16", | |
| "epochs": 1000, | |
| "batch_size": 23, | |
| "eval_batch_size": 23, | |
| "grad_clip": 0.0, | |
| "scheduler_after_epoch": true, | |
| "lr": 5e-06, | |
| "optimizer": "AdamW", | |
| "optimizer_params": { | |
| "betas": [ | |
| 0.9, | |
| 0.96 | |
| ], | |
| "eps": 1e-08, | |
| "weight_decay": 0.01 | |
| }, | |
| "lr_scheduler": "MultiStepLR", | |
| "lr_scheduler_params": { | |
| "milestones": [ | |
| 900000, | |
| 2700000, | |
| 5400000 | |
| ], | |
| "gamma": 0.5, | |
| "last_epoch": -1 | |
| }, | |
| "use_grad_scaler": false, | |
| "allow_tf32": false, | |
| "cudnn_enable": true, | |
| "cudnn_deterministic": false, | |
| "cudnn_benchmark": false, | |
| "training_seed": 1, | |
| "model": "xtts", | |
| "num_loader_workers": 8, | |
| "num_eval_loader_workers": 0, | |
| "use_noise_augment": false, | |
| "audio": { | |
| "sample_rate": 24000, | |
| "output_sample_rate": 24000, | |
| "dvae_sample_rate": 24000 | |
| }, | |
| "use_phonemes": false, | |
| "phonemizer": null, | |
| "phoneme_language": null, | |
| "compute_input_seq_cache": false, | |
| "text_cleaner": null, | |
| "enable_eos_bos_chars": false, | |
| "test_sentences_file": "", | |
| "phoneme_cache_path": null, | |
| "characters": null, | |
| "add_blank": false, | |
| "batch_group_size": 48, | |
| "loss_masking": null, | |
| "min_audio_len": 1, | |
| "max_audio_len": Infinity, | |
| "min_text_len": 1, | |
| "max_text_len": Infinity, | |
| "compute_f0": false, | |
| "compute_energy": false, | |
| "compute_linear_spec": false, | |
| "precompute_num_workers": 0, | |
| "start_by_longest": false, | |
| "shuffle": false, | |
| "drop_last": false, | |
| "datasets": [ | |
| { | |
| "formatter": "", | |
| "dataset_name": "", | |
| "path": "", | |
| "meta_file_train": "", | |
| "ignored_speakers": null, | |
| "language": "", | |
| "phonemizer": "", | |
| "meta_file_val": "", | |
| "meta_file_attn_mask": "" | |
| } | |
| ], | |
| "test_sentences": [ | |
| { | |
| "text": "Ouviram do ipiranga \u00e0s margens pl\u00e1cidas de um povo her\u00f3ico o brado retumbante.", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "Minha terra tem palmeiras onde canta o sabi\u00e1, as aves que aqui gorjeiam n\u00e3o gorjeiam como l\u00e1.", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "\u00d3 que saudades que tenho da aurora da minha vida, da minha inf\u00e2ncia querida, Que os anos n\u00e3o trazem mais.", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "No princ\u00edpio Deus criou o c\u00e9u e a terra, entretanto a terra era sem forma e vazia.", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "Amor \u00e9 fogo que arde sem se ver \u00e9 ferida que d\u00f3i e n\u00e3o se sente \u00e9 um contentamento descontente \u00e9 dor que desatina sem doer.", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "E agora Jos\u00e9? A festa acabou, a luz apagou, o povo sumiu, a noite esfriou, e agora Jos\u00e9?", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "Vou-me embora pra Pas\u00e1rgada, L\u00e1 sou amigo do rei, L\u00e1 tenho a mulher que eu quero, Na cama que escolherei!", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "\u00c9 pau, \u00e9 pedra, \u00e9 o fim do caminho. \u00c9 um resto de toco, \u00e9 um pouco sozinho.", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "No meio do caminho tinha uma pedra; Tinha uma pedra no meio do caminho.", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| }, | |
| { | |
| "text": "Brasil, mostra tua cara; quero ver quem paga pra gente ficar assim.", | |
| "speaker_wav": [ | |
| "/root/DATASETS/BRSpeech_CML_TTS_v14012024_24khz/train/audio/12249/12765/12249_12765_000007-0003.wav" | |
| ], | |
| "language": "pt" | |
| } | |
| ], | |
| "eval_split_max_size": 256, | |
| "eval_split_size": 0.05, | |
| "use_speaker_weighted_sampler": false, | |
| "speaker_weighted_sampler_alpha": 1.0, | |
| "use_language_weighted_sampler": false, | |
| "language_weighted_sampler_alpha": 1.0, | |
| "use_length_weighted_sampler": false, | |
| "length_weighted_sampler_alpha": 1.0, | |
| "model_args": { | |
| "gpt_batch_size": 1, | |
| "enable_redaction": false, | |
| "kv_cache": true, | |
| "gpt_checkpoint": "", | |
| "clvp_checkpoint": null, | |
| "decoder_checkpoint": null, | |
| "num_chars": 255, | |
| "tokenizer_file": "/root/TTS-XTTS-decoder/checkpoints_xtts/XTTS_v2.0_original_model_files/vocab.json", | |
| "gpt_max_audio_tokens": 605, | |
| "gpt_max_text_tokens": 402, | |
| "gpt_max_prompt_tokens": 70, | |
| "gpt_layers": 30, | |
| "gpt_n_model_channels": 1024, | |
| "gpt_n_heads": 16, | |
| "gpt_number_text_tokens": 6681, | |
| "gpt_start_text_token": 261, | |
| "gpt_stop_text_token": 0, | |
| "gpt_num_audio_tokens": 1026, | |
| "gpt_start_audio_token": 1024, | |
| "gpt_stop_audio_token": 1025, | |
| "gpt_code_stride_len": 1024, | |
| "gpt_use_masking_gt_prompt_approach": true, | |
| "gpt_use_perceiver_resampler": true, | |
| "decoder_output_hop_length": 256, | |
| "decoder_input_dim": 1024, | |
| "decoder_d_vector_dim": 512, | |
| "decoder_cond_d_vector_in_each_upsampling_layer": true, | |
| "decoder_resblock_kernel_sizes_decoder": [ | |
| 3, | |
| 7, | |
| 11 | |
| ], | |
| "decoder_resblock_dilation_sizes_decoder": [ | |
| [ | |
| 1, | |
| 3, | |
| 5 | |
| ], | |
| [ | |
| 1, | |
| 3, | |
| 5 | |
| ], | |
| [ | |
| 1, | |
| 3, | |
| 5 | |
| ] | |
| ], | |
| "decoder_upsample_rates_decoder": [ | |
| 8, | |
| 8, | |
| 2, | |
| 2 | |
| ], | |
| "decoder_upsample_initial_channel_decoder": 512, | |
| "decoder_upsample_kernel_sizes_decoder": [ | |
| 16, | |
| 16, | |
| 4, | |
| 4 | |
| ], | |
| "decoder_init_discriminator": false, | |
| "decoder_periods_multi_period_discriminator": [ | |
| 2, | |
| 3, | |
| 5, | |
| 7, | |
| 11 | |
| ], | |
| "decoder_use_spectral_norm_disriminator": false, | |
| "duration_const": 102400, | |
| "min_conditioning_length": 66150, | |
| "max_conditioning_length": 132300, | |
| "gpt_loss_text_ce_weight": 0.01, | |
| "gpt_loss_mel_ce_weight": 1.0, | |
| "debug_loading_failures": false, | |
| "max_wav_length": 255995, | |
| "max_text_length": 200, | |
| "mel_norm_file": "/root/TTS-XTTS-decoder/checkpoints_xtts/XTTS_v2.0_original_model_files/mel_stats.pth", | |
| "dvae_checkpoint": "/root/TTS-XTTS-decoder/checkpoints_xtts/XTTS_v2.0_original_model_files/dvae.pth", | |
| "xtts_checkpoint": "/root/TTS-XTTS-decoder/checkpoints_xtts/GPT_XTTS_v2.0_pt-May-17-2024_01+34AM-3fef64e9/checkpoint_39030.pth", | |
| "vocoder": "" | |
| }, | |
| "model_dir": null, | |
| "languages": [ | |
| "en", | |
| "es", | |
| "fr", | |
| "de", | |
| "it", | |
| "pt", | |
| "pl", | |
| "tr", | |
| "ru", | |
| "nl", | |
| "cs", | |
| "ar", | |
| "zh-cn", | |
| "hu", | |
| "ko", | |
| "ja" | |
| ], | |
| "temperature": 0.85, | |
| "length_penalty": 1.0, | |
| "repetition_penalty": 2.0, | |
| "top_k": 50, | |
| "top_p": 0.85, | |
| "num_gpt_outputs": 1, | |
| "gpt_cond_len": 12, | |
| "gpt_cond_chunk_len": 4, | |
| "max_ref_len": 10, | |
| "sound_norm_refs": false, | |
| "optimizer_wd_only_on_weights": true, | |
| "weighted_loss_attrs": {}, | |
| "weighted_loss_multipliers": {}, | |
| "github_branch": "* main" | |
| } |