config.json

{
  "format": "songgeneration-mlx-audiolm",
  "source": "tencent/SongGeneration",
  "official_code": "https://github.com/tencent-ailab/songgeneration",
  "variant": "v2-medium",
  "precision": "bfloat16",
  "runtime": {
    "hidden_size": 1536,
    "intermediate_size": 8960,
    "num_heads": 12,
    "num_layers": 28,
    "num_layers_sub": 12,
    "code_depth": 3,
    "code_size": 16384,
    "prompt_len": 10,
    "frame_rate": 25,
    "max_position_embeddings": 10000,
    "max_position_embeddings_sub": 10000,
    "rope_theta": 500000.0,
    "rope_theta_sub": 500000.0,
    "cfg_coef": 1.5
  },
  "components": {
    "audiolm": "converted to MLX safetensors",
    "qwen_tokenizer": "copied for lyric/style tokenization",
    "audio_decoder": "use official PyTorch Flow1dVAE/separate tokenizer bridge for now"
  },
  "skipped_keys": [
    "audiolm.out_norm.weight",
    "audiolm.out_norm.bias",
    "audiolm.transformer2.lm_head.weight"
  ],
  "weight_files": [
    "model-00001-of-00094.safetensors",
    "model-00002-of-00094.safetensors",
    "model-00003-of-00094.safetensors",
    "model-00004-of-00094.safetensors",
    "model-00005-of-00094.safetensors",
    "model-00006-of-00094.safetensors",
    "model-00007-of-00094.safetensors",
    "model-00008-of-00094.safetensors",
    "model-00009-of-00094.safetensors",
    "model-00010-of-00094.safetensors",
    "model-00011-of-00094.safetensors",
    "model-00012-of-00094.safetensors",
    "model-00013-of-00094.safetensors",
    "model-00014-of-00094.safetensors",
    "model-00015-of-00094.safetensors",
    "model-00016-of-00094.safetensors",
    "model-00017-of-00094.safetensors",
    "model-00018-of-00094.safetensors",
    "model-00019-of-00094.safetensors",
    "model-00020-of-00094.safetensors",
    "model-00021-of-00094.safetensors",
    "model-00022-of-00094.safetensors",
    "model-00023-of-00094.safetensors",
    "model-00024-of-00094.safetensors",
    "model-00025-of-00094.safetensors",
    "model-00026-of-00094.safetensors",
    "model-00027-of-00094.safetensors",
    "model-00028-of-00094.safetensors",
    "model-00029-of-00094.safetensors",
    "model-00030-of-00094.safetensors",
    "model-00031-of-00094.safetensors",
    "model-00032-of-00094.safetensors",
    "model-00033-of-00094.safetensors",
    "model-00034-of-00094.safetensors",
    "model-00035-of-00094.safetensors",
    "model-00036-of-00094.safetensors",
    "model-00037-of-00094.safetensors",
    "model-00038-of-00094.safetensors",
    "model-00039-of-00094.safetensors",
    "model-00040-of-00094.safetensors",
    "model-00041-of-00094.safetensors",
    "model-00042-of-00094.safetensors",
    "model-00043-of-00094.safetensors",
    "model-00044-of-00094.safetensors",
    "model-00045-of-00094.safetensors",
    "model-00046-of-00094.safetensors",
    "model-00047-of-00094.safetensors",
    "model-00048-of-00094.safetensors",
    "model-00049-of-00094.safetensors",
    "model-00050-of-00094.safetensors",
    "model-00051-of-00094.safetensors",
    "model-00052-of-00094.safetensors",
    "model-00053-of-00094.safetensors",
    "model-00054-of-00094.safetensors",
    "model-00055-of-00094.safetensors",
    "model-00056-of-00094.safetensors",
    "model-00057-of-00094.safetensors",
    "model-00058-of-00094.safetensors",
    "model-00059-of-00094.safetensors",
    "model-00060-of-00094.safetensors",
    "model-00061-of-00094.safetensors",
    "model-00062-of-00094.safetensors",
    "model-00063-of-00094.safetensors",
    "model-00064-of-00094.safetensors",
    "model-00065-of-00094.safetensors",
    "model-00066-of-00094.safetensors",
    "model-00067-of-00094.safetensors",
    "model-00068-of-00094.safetensors",
    "model-00069-of-00094.safetensors",
    "model-00070-of-00094.safetensors",
    "model-00071-of-00094.safetensors",
    "model-00072-of-00094.safetensors",
    "model-00073-of-00094.safetensors",
    "model-00074-of-00094.safetensors",
    "model-00075-of-00094.safetensors",
    "model-00076-of-00094.safetensors",
    "model-00077-of-00094.safetensors",
    "model-00078-of-00094.safetensors",
    "model-00079-of-00094.safetensors",
    "model-00080-of-00094.safetensors",
    "model-00081-of-00094.safetensors",
    "model-00082-of-00094.safetensors",
    "model-00083-of-00094.safetensors",
    "model-00084-of-00094.safetensors",
    "model-00085-of-00094.safetensors",
    "model-00086-of-00094.safetensors",
    "model-00087-of-00094.safetensors",
    "model-00088-of-00094.safetensors",
    "model-00089-of-00094.safetensors",
    "model-00090-of-00094.safetensors",
    "model-00091-of-00094.safetensors",
    "model-00092-of-00094.safetensors",
    "model-00093-of-00094.safetensors",
    "model-00094-of-00094.safetensors"
  ]
}