{
  "base_model": "microsoft/VibeVoice-1.5B",
  "checkpoint_dir": "/kaggle/working/vibevoice-finetune/out_tpu/final",
  "dtype": "bfloat16",
  "components_merged": [
    "language_model",
    "prediction_head",
    "acoustic_connector",
    "semantic_connector"
  ],
  "tensors_changed_vs_base": 373,
  "training_state": {
    "step": 2508,
    "args": {
      "model_name_or_path": "microsoft/VibeVoice-1.5B",
      "data_dir": "/kaggle/working/preprocessed/train",
      "eval_data_dir": null,
      "output_dir": "out_tpu",
      "lazy_data": false,
      "max_seq_len": 0,
      "max_latents": 0,
      "max_segments": 0,
      "buckets": null,
      "auto_buckets": 0,
      "per_device_batch_size": 2,
      "gradient_accumulation_steps": 4,
      "learning_rate": 1e-05,
      "head_lr_multiplier": 2.0,
      "weight_decay": 0.01,
      "adam_beta1": 0.9,
      "adam_beta2": 0.95,
      "adam_eps": 1e-08,
      "max_grad_norm": 1.0,
      "no_master_weights": false,
      "num_train_epochs": 6.0,
      "max_steps": 0,
      "warmup_ratio": 0.08,
      "warmup_steps": 0,
      "lr_scheduler": "cosine",
      "min_lr_ratio": 0.05,
      "seed": 42,
      "ce_loss_weight": 1.2,
      "diffusion_loss_weight": 1.0,
      "ddpm_batch_mul": 6,
      "ce_chunk_size": 128,
      "no_semantic": false,
      "no_latent_noise": false,
      "ce_acoustic_supervision": "target",
      "freeze_components": null,
      "freeze_llm_layers": null,
      "train_llm_layers": null,
      "freeze_head_layers": null,
      "freeze_embeddings": false,
      "freeze_lm_head": false,
      "freeze_regex": null,
      "bf16": true,
      "gradient_checkpointing": true,
      "no_spmd": false,
      "logging_steps": 10,
      "save_steps": 2000,
      "save_total_limit": 1,
      "eval_steps": 0,
      "eval_batches": 20,
      "resume_from_checkpoint": null,
      "dataloader_workers": 0,
      "profile_first_steps": 3
    },
    "shapes": {
      "max_seq_len": 512,
      "max_latents": 224,
      "max_segments": 8,
      "acoustic_dim": 64,
      "semantic_dim": 128
    },
    "preprocess_metadata": {
      "model_name_or_path": "microsoft/VibeVoice-1.5B",
      "processor_name_or_path": "vibevoice/processor",
      "compress_ratio": 3200,
      "acoustic_dim": 64,
      "semantic_dim": 128,
      "fix_std": 0.5,
      "latent_dtype": "float16",
      "speech_scaling_factor": 0.1962890625,
      "speech_bias_factor": -0.04931640625,
      "scaling_source": "checkpoint",
      "corpus_latent_mean": -0.16465343947872957,
      "corpus_latent_var": 26.317665766460568,
      "shapes": {
        "max_seq_len": 512,
        "max_latents": 224,
        "max_segments": 8,
        "acoustic_dim": 64,
        "semantic_dim": 128
      },
      "pad_token_id": 151655,
      "augment_target_silence": true,
      "normalize_target_audio": false
    }
  }
}