Upload 2 files
Browse filesFine-tune the SeedVC model (seed-uvit-whisper-small-wavenet) for the Ukrainian language. The model no longer has an English accent and understands Ukrainian well. The original model’s features have been preserved. It was trained on private data featuring various voices. Approximately 1,000 files for training.
config_dit_mel_seed_uvit_whisper_small_wavenet.yml
ADDED
|
@@ -0,0 +1,91 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
log_dir: "./runs"
|
| 2 |
+
save_freq: 1
|
| 3 |
+
log_interval: 10
|
| 4 |
+
save_interval: 1000
|
| 5 |
+
device: "cuda"
|
| 6 |
+
epochs: 1000 # number of epochs for first stage training (pre-training)
|
| 7 |
+
batch_size: 2
|
| 8 |
+
batch_length: 100 # maximum duration of audio in a batch (in seconds)
|
| 9 |
+
max_len: 80 # maximum number of frames
|
| 10 |
+
pretrained_model: "DiT_seed_v2_uvit_whisper_small_wavenet_bigvgan_pruned.pth"
|
| 11 |
+
pretrained_encoder: ""
|
| 12 |
+
load_only_params: False # set to true if do not want to load epoch numbers and optimizer parameters
|
| 13 |
+
|
| 14 |
+
preprocess_params:
|
| 15 |
+
sr: 22050
|
| 16 |
+
spect_params:
|
| 17 |
+
n_fft: 1024
|
| 18 |
+
win_length: 1024
|
| 19 |
+
hop_length: 256
|
| 20 |
+
n_mels: 80
|
| 21 |
+
fmin: 0
|
| 22 |
+
fmax: "None"
|
| 23 |
+
|
| 24 |
+
model_params:
|
| 25 |
+
dit_type: "DiT" # uDiT or DiT
|
| 26 |
+
reg_loss_type: "l1" # l1 or l2
|
| 27 |
+
|
| 28 |
+
timbre_shifter:
|
| 29 |
+
se_db_path: "./modules/openvoice/checkpoints_v2/converter/se_db.pt"
|
| 30 |
+
ckpt_path: './modules/openvoice/checkpoints_v2/converter'
|
| 31 |
+
|
| 32 |
+
speech_tokenizer:
|
| 33 |
+
type: 'whisper'
|
| 34 |
+
name: "openai/whisper-small"
|
| 35 |
+
|
| 36 |
+
style_encoder:
|
| 37 |
+
dim: 192
|
| 38 |
+
campplus_path: "campplus_cn_common.bin"
|
| 39 |
+
|
| 40 |
+
vocoder:
|
| 41 |
+
type: "bigvgan"
|
| 42 |
+
name: "nvidia/bigvgan_v2_22khz_80band_256x"
|
| 43 |
+
|
| 44 |
+
length_regulator:
|
| 45 |
+
channels: 512
|
| 46 |
+
is_discrete: false
|
| 47 |
+
in_channels: 768
|
| 48 |
+
content_codebook_size: 2048
|
| 49 |
+
sampling_ratios: [1, 1, 1, 1]
|
| 50 |
+
vector_quantize: false
|
| 51 |
+
n_codebooks: 1
|
| 52 |
+
quantizer_dropout: 0.0
|
| 53 |
+
f0_condition: false
|
| 54 |
+
n_f0_bins: 512
|
| 55 |
+
|
| 56 |
+
DiT:
|
| 57 |
+
hidden_dim: 512
|
| 58 |
+
num_heads: 8
|
| 59 |
+
depth: 13
|
| 60 |
+
class_dropout_prob: 0.1
|
| 61 |
+
block_size: 8192
|
| 62 |
+
in_channels: 80
|
| 63 |
+
style_condition: true
|
| 64 |
+
final_layer_type: 'wavenet'
|
| 65 |
+
target: 'mel' # mel or codec
|
| 66 |
+
content_dim: 512
|
| 67 |
+
content_codebook_size: 1024
|
| 68 |
+
content_type: 'discrete'
|
| 69 |
+
f0_condition: false
|
| 70 |
+
n_f0_bins: 512
|
| 71 |
+
content_codebooks: 1
|
| 72 |
+
is_causal: false
|
| 73 |
+
long_skip_connection: true
|
| 74 |
+
zero_prompt_speech_token: false # for prompt component, do not input corresponding speech token
|
| 75 |
+
time_as_token: false
|
| 76 |
+
style_as_token: false
|
| 77 |
+
uvit_skip_connection: true
|
| 78 |
+
add_resblock_in_transformer: false
|
| 79 |
+
|
| 80 |
+
wavenet:
|
| 81 |
+
hidden_dim: 512
|
| 82 |
+
num_layers: 8
|
| 83 |
+
kernel_size: 5
|
| 84 |
+
dilation_rate: 1
|
| 85 |
+
p_dropout: 0.2
|
| 86 |
+
style_condition: true
|
| 87 |
+
|
| 88 |
+
loss_params:
|
| 89 |
+
base_lr: 0.0001
|
| 90 |
+
lambda_mel: 45
|
| 91 |
+
lambda_kl: 1.0
|
ft_model_ukr.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1fd3d3923c1fdfe49c75473767cbe2bd61193c30bf0a6e971ce528cd3ad10acb
|
| 3 |
+
size 412426080
|