Sucial commited on
Commit
05a8bb1
1 Parent(s): ee050c3

Upload 6 files

Browse files
Files changed (6) hide show
  1. bark.json +107 -0
  2. bark.pth +3 -0
  3. diffusion.pt +3 -0
  4. diffusion.yaml +53 -0
  5. feature_and_index.pkl +3 -0
  6. kmeans.pt +3 -0
bark.json ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train": {
3
+ "log_interval": 200,
4
+ "eval_interval": 800,
5
+ "seed": 1234,
6
+ "epochs": 10000,
7
+ "learning_rate": 0.0004,
8
+ "betas": [
9
+ 0.8,
10
+ 0.99
11
+ ],
12
+ "eps": 1e-09,
13
+ "batch_size": 16,
14
+ "fp16_run": false,
15
+ "half_type": "fp16",
16
+ "lr_decay": 0.999875,
17
+ "segment_size": 10240,
18
+ "init_lr_ratio": 1,
19
+ "warmup_epochs": 0,
20
+ "c_mel": 45,
21
+ "c_kl": 1.0,
22
+ "use_sr": true,
23
+ "max_speclen": 512,
24
+ "port": "8001",
25
+ "keep_ckpts": 0,
26
+ "all_in_mem": false,
27
+ "vol_aug": true
28
+ },
29
+ "data": {
30
+ "training_files": "filelists/train.txt",
31
+ "validation_files": "filelists/val.txt",
32
+ "max_wav_value": 32768.0,
33
+ "sampling_rate": 44100,
34
+ "filter_length": 2048,
35
+ "hop_length": 512,
36
+ "win_length": 2048,
37
+ "n_mel_channels": 80,
38
+ "mel_fmin": 0.0,
39
+ "mel_fmax": 22050,
40
+ "unit_interpolate_mode": "nearest"
41
+ },
42
+ "model": {
43
+ "inter_channels": 192,
44
+ "hidden_channels": 192,
45
+ "filter_channels": 768,
46
+ "n_heads": 2,
47
+ "n_layers": 6,
48
+ "kernel_size": 3,
49
+ "p_dropout": 0.1,
50
+ "resblock": "1",
51
+ "resblock_kernel_sizes": [
52
+ 3,
53
+ 7,
54
+ 11
55
+ ],
56
+ "resblock_dilation_sizes": [
57
+ [
58
+ 1,
59
+ 3,
60
+ 5
61
+ ],
62
+ [
63
+ 1,
64
+ 3,
65
+ 5
66
+ ],
67
+ [
68
+ 1,
69
+ 3,
70
+ 5
71
+ ]
72
+ ],
73
+ "upsample_rates": [
74
+ 8,
75
+ 8,
76
+ 2,
77
+ 2,
78
+ 2
79
+ ],
80
+ "upsample_initial_channel": 512,
81
+ "upsample_kernel_sizes": [
82
+ 16,
83
+ 16,
84
+ 4,
85
+ 4,
86
+ 4
87
+ ],
88
+ "n_layers_q": 3,
89
+ "n_flow_layer": 4,
90
+ "use_spectral_norm": false,
91
+ "gin_channels": 768,
92
+ "ssl_dim": 768,
93
+ "n_speakers": 3,
94
+ "vocoder_name": "nsf-hifigan",
95
+ "speech_encoder": "vec768l12",
96
+ "speaker_embedding": false,
97
+ "vol_embedding": true,
98
+ "use_depthwise_conv": false,
99
+ "flow_share_parameter": false,
100
+ "use_automatic_f0_prediction": true
101
+ },
102
+ "spk": {
103
+ "ferocious": 0,
104
+ "husky": 1,
105
+ "puppy": 2
106
+ }
107
+ }
bark.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ea95a730092bae5145d8619eac9dae7787a2bd6126ff0263005e093ff411af4
3
+ size 160943725
diffusion.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a0d6eeee43658c6afb9d8553f999f9bfb59cb3a61bdbed513dd428f4bc4372f8
3
+ size 220897161
diffusion.yaml ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data:
2
+ block_size: 512
3
+ cnhubertsoft_gate: 10
4
+ duration: 2
5
+ encoder: vec768l12
6
+ encoder_hop_size: 320
7
+ encoder_out_channels: 768
8
+ encoder_sample_rate: 16000
9
+ extensions:
10
+ - wav
11
+ sampling_rate: 44100
12
+ training_files: filelists/train.txt
13
+ unit_interpolate_mode: nearest
14
+ validation_files: filelists/val.txt
15
+ device: cuda
16
+ env:
17
+ expdir: logs/44k/diffusion
18
+ gpu_id: 0
19
+ infer:
20
+ method: dpm-solver++
21
+ speedup: 10
22
+ model:
23
+ k_step_max: 0
24
+ n_chans: 512
25
+ n_hidden: 256
26
+ n_layers: 20
27
+ n_spk: 3
28
+ timesteps: 1000
29
+ type: Diffusion
30
+ use_pitch_aug: true
31
+ spk:
32
+ ferocious: 0
33
+ husky: 1
34
+ puppy: 2
35
+ train:
36
+ amp_dtype: fp32
37
+ batch_size: 48
38
+ cache_all_data: true
39
+ cache_device: cpu
40
+ cache_fp16: true
41
+ decay_step: 100000
42
+ epochs: 100000
43
+ gamma: 0.5
44
+ interval_force_save: 5000
45
+ interval_log: 10
46
+ interval_val: 2000
47
+ lr: 0.0001
48
+ num_workers: 4
49
+ save_opt: false
50
+ weight_decay: 0
51
+ vocoder:
52
+ ckpt: pretrain/nsf_hifigan/model
53
+ type: nsf-hifigan
feature_and_index.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d94ebcec894746f5506dae365c7e898632c266dc645295aee4a6ef468665c0f4
3
+ size 94026795
kmeans.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:246f46790f42a354d5162f97b0e4a119ac04bbf9272d3eb9a508cd8908880488
3
+ size 93411321