train_config.json

{
  "data_dir": "data/tokenized_siwis,data/tokenized_siwis_full",
  "val_dir": "data/tokenized_val_full",
  "output_dir": "outputs/full_ft_vibevoice",
  "epochs": 10,
  "batch_size": 4,
  "grad_accum": 4,
  "lr": 5e-05,
  "weight_decay": 0.01,
  "warmup_steps": 500,
  "full_finetune": true,
  "lora_rank": 16,
  "lora_alpha": 32.0,
  "max_text_len": 256,
  "max_speech_len": 120,
  "device": "cuda",
  "bf16": true,
  "save_every": 500,
  "eval_every": 500
}