| { | |
| "architecture": "voxcpm", | |
| "lm_config": { | |
| "bos_token_id": 1, | |
| "eos_token_id": 2, | |
| "hidden_size": 1024, | |
| "intermediate_size": 4096, | |
| "max_position_embeddings": 32768, | |
| "num_attention_heads": 16, | |
| "num_hidden_layers": 24, | |
| "num_key_value_heads": 2, | |
| "rms_norm_eps": 1e-05, | |
| "rope_theta": 10000, | |
| "rope_scaling": { | |
| "type": "longrope", | |
| "long_factor": [1.0004360675811768, 1.0668443441390991, 1.1631425619125366, 1.3025742769241333, 1.5040205717086792, 1.7941505908966064, 2.2101221084594727, 2.802666664123535, 3.6389970779418945, 4.804192543029785, 6.39855432510376, 8.527148246765137, 11.277542114257812, 14.684998512268066, 18.69317054748535, 23.13019371032715, 27.72362518310547, 32.1606559753418, 36.168827056884766, 39.57627868652344, 42.32667541503906, 44.45526885986328, 46.04962921142578, 47.21482849121094, 48.05115509033203, 48.64370346069336, 49.05967712402344, 49.34980392456055, 49.551246643066406, 49.69068145751953, 49.78697967529297, 49.85338592529297], | |
| "short_factor": [1.0004360675811768, 1.0668443441390991, 1.1631425619125366, 1.3025742769241333, 1.5040205717086792, 1.7941505908966064, 2.2101221084594727, 2.802666664123535, 3.6389970779418945, 4.804192543029785, 6.39855432510376, 8.527148246765137, 11.277542114257812, 14.684998512268066, 18.69317054748535, 23.13019371032715, 27.72362518310547, 32.1606559753418, 36.168827056884766, 39.57627868652344, 42.32667541503906, 44.45526885986328, 46.04962921142578, 47.21482849121094, 48.05115509033203, 48.64370346069336, 49.05967712402344, 49.34980392456055, 49.551246643066406, 49.69068145751953, 49.78697967529297, 49.85338592529297], | |
| "original_max_position_embeddings": 32768 | |
| }, | |
| "vocab_size": 73448, | |
| "scale_emb": 12, | |
| "dim_model_base": 256, | |
| "scale_depth": 1.4, | |
| "use_mup": false | |
| }, | |
| "patch_size": 4, | |
| "feat_dim": 64, | |
| "scalar_quantization_latent_dim": 256, | |
| "scalar_quantization_scale": 9, | |
| "residual_lm_num_layers": 8, | |
| "encoder_config": { | |
| "hidden_dim": 1024, | |
| "ffn_dim": 4096, | |
| "num_heads": 16, | |
| "num_layers": 8 | |
| }, | |
| "dit_config": { | |
| "hidden_dim": 1024, | |
| "ffn_dim": 4096, | |
| "num_heads": 16, | |
| "num_layers": 8, | |
| "cfm_config": { | |
| "sigma_min": 1e-06, | |
| "solver": "euler", | |
| "t_scheduler": "log-norm", | |
| "inference_cfg_rate": 2.0 | |
| } | |
| }, | |
| "audio_vae_config": { | |
| "encoder_dim": 64, | |
| "encoder_rates": [2, 3, 6, 7, 7], | |
| "latent_dim": 64, | |
| "decoder_dim": 2048, | |
| "decoder_rates": [7, 7, 6, 3, 2], | |
| "sample_rate": 44100 | |
| }, | |
| "max_length": 8192, | |
| "device": "cuda", | |
| "dtype": "bfloat16" | |
| } | |