| { |
| "dim": 640, |
| "d_model": 640, |
| "n_layers": 20, |
| "num_layers": 20, |
| "n_heads": 10, |
| "num_heads": 10, |
| "d_head": 64, |
| "d_k": 64, |
| "d_v": 64, |
| "vocab_size": 50257, |
| "max_seq_len": 2048, |
| "d_emb": 128, |
| "kernel_size": 4, |
| "conv_kernel_size": 4, |
| "chunk_size": 16, |
| "eps": 1e-06, |
| "inversion_method": "adaptive", |
| "inversion_threshold": 0.5, |
| "adaptive_tol": 7e-05, |
| "d_c": 128, |
| "d_idx": 64, |
| "block_size": 64, |
| "window_size": 128, |
| "top_k": 32, |
| "hca_pool_size": 64, |
| "dist_lambda": 0.5, |
| "intermediate_size": 1248, |
| "rms_norm_eps": 1e-06, |
| "residual_gate_bias": 2.0, |
| "mtp_depth": 2, |
| "architectures": [ |
| "MabaSparseForCausalLM" |
| ], |
| "model_type": "maba_sparse", |
| "torch_dtype": "bfloat16", |
| "auto_map": { |
| "AutoConfig": "maba_sparse.config.MabaSparseConfig", |
| "AutoModelForCausalLM": "maba_sparse.model.MabaSparseForCausalLM" |
| } |
| } |
