File size: 1,299 Bytes
9834244 d1c85f4 9834244 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | # Complete EMA model; inference only.
model:
name: WaveGen_Flash
tokenizer: pinyin
backbone: DiT
cfm:
prediction: x_pred
t_eps: 0.02
latents_scale: 9.0
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 # Disabled explicitly by inference_bottleneck_dropout below.
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
inference:
num_steps: 4
sampling_method: euler
time_shift: 3.0
# Already shifted: t / (t + 3 * (1 - t)), from [0, .25, .5, .75, 1].
# Use these exact evaluation times; do not apply another time transform.
time_grid: [0.0, 0.1, 0.25, 0.5, 1.0]
cfg_strength: 0.0
inference_bottleneck_dropout: 0.0
waveform_temperature: 1.0
# Initial Mel noise only; use 1.0 to reproduce the previous setting.
mel_temperature: 0.0
compute_final_mel: false
max_duration_seconds: 30.0
|