train_data:
  train_audio_sample_rate: 48000
  audio_samples_per_llm_token: 7680
  sources:
    - name: ljspeech_meanflow_basic
      weight: 1.0
      pipeline: basic
      adapter:
        class_name: JsonlManifestSourceAdapter
        params:
          manifest_path: downloaded_data/ljspeech_48khz_manifest_train.jsonl
          shuffle: true
    # append other sources here if need
  num_tokens_per_epoch: 1000000
  num_workers: 4
  pin_memory: true
  max_audio_seconds_in_batch: 10.0
  max_text_tokens_in_batch: 1024
  max_samples_per_batch: 1
  bucketing_pool_size: 64
val_data:
  train_audio_sample_rate: 48000
  audio_samples_per_llm_token: 7680
  sources:
    - name: ljspeech_meanflow_valid_basic
      weight: 1.0
      pipeline: basic
      adapter:
        class_name: JsonlManifestSourceAdapter
        params:
          manifest_path: downloaded_data/ljspeech_48khz_manifest_valid.jsonl
          shuffle: false
    # append other sources here if need
  num_workers: 2
  pin_memory: true
  max_audio_seconds_in_batch: 10.0
  max_text_tokens_in_batch: 1024
  max_samples_per_batch: 1
  bucketing_pool_size: 32
train:
  pretrained_model_path: pretrained_models/dots.tts-soar
  output_dir: debug_train/meanflow_run_001
  seed: 42
  learning_rate: 1.0e-05
  cfg_droprate: 0.0
  xvec_drop_rate: 0.5
  weight_decay: 0.01
  warmup_steps: 50
  max_train_steps: 500
  gradient_accumulation_steps: 1
  grad_clip_norm: 1
  save_interval: 500
  max_checkpoints_to_keep: 10
  log_interval: 10
  eval_interval: 100
  max_eval_batches: 10
  run_eval_on_start: false
loss:
  ce_weight: 1.0
  fm_weight: 1.0
  eos_weight: 1.0