| { |
| "architectures": [ |
| "MuseSparkForConditionalGeneration" |
| ], |
| "audio_config": { |
| "conv_kernel_size": 7, |
| "depths": [ |
| 2, |
| 2, |
| 2, |
| 2 |
| ], |
| "hidden_act": "gelu", |
| "hidden_sizes": [ |
| 8, |
| 16, |
| 32, |
| 64 |
| ], |
| "initializer_range": 0.02, |
| "mlp_ratio": 4.0, |
| "model_type": "muse_spark_audio", |
| "num_hidden_layers": 4, |
| "out_hidden_size": 8, |
| "patch_sizes": [ |
| 16, |
| 6, |
| 5, |
| 4 |
| ], |
| "rms_norm_eps": 1e-06 |
| }, |
| "audio_token_id": 200038, |
| "dtype": "bfloat16", |
| "image_token_id": 200092, |
| "model_type": "muse_spark", |
| "out_hidden_size": 128, |
| "projector_hidden_act": "gelu", |
| "projector_hidden_size": 16, |
| "text_config": { |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 200000, |
| "eos_token_id": 200001, |
| "final_logit_softcapping": 20.0, |
| "head_dim": 32, |
| "hidden_act": "silu", |
| "hidden_size": 8, |
| "initializer_range": 0.02, |
| "intermediate_size": 32, |
| "layer_rope_theta": [ |
| 500000.0, |
| 0, |
| 500000.0, |
| 500000.0, |
| 500000.0, |
| 0, |
| 500000.0, |
| 500000.0 |
| ], |
| "layer_types": [ |
| "sliding_attention", |
| "full_attention", |
| "sliding_attention", |
| "sliding_attention", |
| "sliding_attention", |
| "full_attention", |
| "sliding_attention", |
| "sliding_attention" |
| ], |
| "max_position_embeddings": 1048576, |
| "mlp_layer_types": [ |
| "sparse", |
| "sparse", |
| "sparse", |
| "sparse", |
| "sparse", |
| "sparse", |
| "sparse", |
| "sparse" |
| ], |
| "model_type": "muse_spark_text", |
| "moe_hidden_size": 4, |
| "moe_intermediate_size": 16, |
| "num_attention_heads": 8, |
| "num_experts_per_tok": 8, |
| "num_hidden_layers": 8, |
| "num_key_value_heads": 1, |
| "num_local_experts": 32, |
| "num_shared_experts": 0, |
| "output_multiplier": 0.17677669529663687, |
| "pad_token_id": 200018, |
| "post_norm_eps": 1e-08, |
| "qk_scale_factor": 5.4730064863875, |
| "residual_gate_temperature": 0.3, |
| "rms_norm_eps": 1e-05, |
| "rope_parameters": { |
| "rope_theta": 500000.0, |
| "rope_type": "default" |
| }, |
| "routed_scaling_factor": 1.0, |
| "sliding_window": 2048, |
| "tie_word_embeddings": false, |
| "use_cache": true, |
| "vocab_size": 202048 |
| }, |
| "transformers_version": "5.18.0.dev0", |
| "video_token_id": 200091, |
| "vision_config": { |
| "hidden_act": "gelu", |
| "hidden_size": 32, |
| "intermediate_size": 64, |
| "layer_norm_eps": 1e-05, |
| "layer_types": [ |
| "window_attention", |
| "window_attention", |
| "window_attention", |
| "full_attention", |
| "window_attention", |
| "window_attention", |
| "window_attention", |
| "full_attention" |
| ], |
| "max_position_embeddings": 1024, |
| "merge_size": 2, |
| "model_type": "muse_spark_vision", |
| "num_attention_heads": 4, |
| "num_hidden_layers": 8, |
| "patch_size": 14, |
| "patch_temporal": 2, |
| "pos_emb_height": 32, |
| "pos_emb_width": 32, |
| "rope_parameters": { |
| "rope_theta": 10000, |
| "rope_type": "axial" |
| } |
| } |
| } |
|
|
