Code-Dev commited on
Commit
69cdf97
·
verified ·
1 Parent(s): fd3e7ee

Initialize clean base with manassehzw Phase 1 Shona Male F5-TTS weights

Browse files
Files changed (1) hide show
  1. config.yaml +44 -0
config.yaml ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exp_name: shona_f5_tts-m150-n150-full-lr1em5-ep30-April-26-2026_02+51PM-b60c1d4
2
+ output_dir: /root/project/runs/shona_f5_tts/shona_f5_tts-m150-n150-full-lr1em5-ep30-April-26-2026_02+51PM-b60c1d4/checkpoints
3
+ seed: 42
4
+ model:
5
+ type: F5TTS_v1_Base
6
+ pretrained: SWivid/F5-TTS
7
+ pretrained_ckpt: F5TTS_v1_Base/model_1250000.safetensors
8
+ vocab_path: /root/project/datasets/sna-manasseh-150-raw/f5_prepared/default/vocab.txt
9
+ use_lora: false
10
+ dim: 1024
11
+ depth: 22
12
+ heads: 16
13
+ ff_mult: 2
14
+ text_dim: 512
15
+ conv_layers: 4
16
+ dataset:
17
+ name: shona_f5_tts-m150-n150-full-lr1em5-ep30-April-26-2026_02+51PM-b60c1d4
18
+ dir: /root/project/datasets/sna-manasseh-150-raw/f5_prepared/default
19
+ metadata: metadata.csv
20
+ sample_rate: 24000
21
+ training:
22
+ epochs: 30
23
+ learning_rate: 1.0e-05
24
+ weight_decay: 0.01
25
+ max_grad_norm: 1.0
26
+ grad_accumulation_steps: 1
27
+ batch_size_per_gpu: 1600
28
+ batch_size_type: frame
29
+ max_samples: 64
30
+ num_warmup_updates: 500
31
+ lr_scheduler: cosine
32
+ use_ema: true
33
+ ema_decay: 0.9999
34
+ mixed_precision: fp16
35
+ num_workers: 4
36
+ checkpoint:
37
+ save_per_updates: 66
38
+ last_per_steps: 10000
39
+ keep_last_n: 5
40
+ logging:
41
+ type: wandb
42
+ log_per_updates: 50
43
+ wandb_project: f5-tts-shona-finetuning
44
+ wandb_run_name: shona_f5_tts-m150-n150-full-lr1em5-ep30-April-26-2026_02+51PM-b60c1d4