| model: | |
| name: WaveGen_Emilia_24k_50Hz_SF_80bin_ibdim32_drop02 | |
| tokenizer: pinyin | |
| backbone: DiT | |
| cfm: | |
| prediction: x_pred | |
| loss_space: v | |
| t_sampling: logistic_normal | |
| P_mean: -0.8 | |
| P_std: 0.8 | |
| time_shift: 1.0 | |
| t_eps: 0.02 | |
| latents_scale: 9.0 | |
| use_aux_mel_loss: false | |
| aux_mel_loss_weight: 0.0 | |
| use_ms_v_spec_loss: true | |
| ms_v_spec_loss_weight: 0.1 | |
| aux_mel_loss_masked: true | |
| sf_loss_weight: 0.1 | |
| mel_xt_drop_prob: 0.0 | |
| inference_bottleneck_dropout: 0.2 | |
| inference_bottleneck_dropout_seed: 0 | |
| sample_rate: 24000 | |
| joint_cond_drop_prob: 0.1 | |
| arch: | |
| dim: 1152 | |
| depth: 24 | |
| heads: 16 | |
| dropout: 0.0 | |
| ff_mult: 4 | |
| text_dim: 512 | |
| text_mask_padding: true | |
| qk_norm: null | |
| conv_layers: 4 | |
| pe_attn_head: null | |
| attn_backend: torch | |
| attn_mask_enabled: false | |
| checkpoint_activations: false | |
| mel_channels: 80 | |
| use_mel_information_bottleneck: true | |
| mel_bottleneck_dim: 32 | |
| mel_bottleneck_dropout: 0.2 | |
| mel_bottleneck_zero_init: false | |
| use_audio_proj: true | |
| audio_proj_dim: 1024 | |
| audio_proj_hidden: 2048 | |
| waveform: | |
| target_sample_rate: 24000 | |
| wav_frame_len: 480 | |