Daniel O'Neill commited on
Commit
cea5a8e
1 Parent(s): de2f105

Add PrimReaper model

Browse files
PrimReaper/config.yaml ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data:
2
+ block_size: 512
3
+ cnhubertsoft_gate: 10
4
+ duration: 2
5
+ encoder: contentvec768l12
6
+ encoder_ckpt: pretrain/contentvec/checkpoint_best_legacy_500.pt
7
+ encoder_hop_size: 320
8
+ encoder_out_channels: 768
9
+ encoder_sample_rate: 16000
10
+ extensions:
11
+ - wav
12
+ f0_extractor: rmvpe
13
+ f0_max: 800
14
+ f0_min: 65
15
+ sampling_rate: 44100
16
+ train_path: data/train
17
+ valid_path: data/val
18
+ device: cuda
19
+ env:
20
+ expdir: exp/diffusion-test-16-2
21
+ gpu_id: 1
22
+ infer:
23
+ method: dpm-solver
24
+ speedup: 10
25
+ model:
26
+ k_step_max: 100
27
+ n_chans: 512
28
+ n_layers: 10
29
+ n_spk: 1
30
+ pcmer_norm: true
31
+ type: DiffusionNew
32
+ use_pitch_aug: true
33
+ train:
34
+ amp_dtype: fp16
35
+ batch_size: 36
36
+ cache_all_data: true
37
+ cache_device: cuda
38
+ cache_fp16: true
39
+ decay_step: 25000
40
+ epochs: 100000
41
+ gamma: 0.5
42
+ interval_force_save: 10000
43
+ interval_log: 1
44
+ interval_val: 2000
45
+ lambda_ddsp: 1
46
+ lr: 0.0002
47
+ num_workers: 0
48
+ save_opt: false
49
+ weight_decay: 0
50
+ vocoder:
51
+ ckpt: pretrain/nsf_hifigan/model
52
+ type: nsf-hifigan
PrimReaper/model_50000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99dc65df5b1059339885078df048afbfaafb99054baae361b4a3c00ef5af9072
3
+ size 131310766