| { | |
| "_name_or_path": "MiniMaxAI/MiniMax-Text-01", | |
| "architectures": [ | |
| "MiniMaxText01ForCausalLM" | |
| ], | |
| "attention_dropout": 0.0, | |
| "attn_type_list": [ | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| 1 | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "MiniMaxAI/MiniMax-Text-01--configuration_minimax_text_01.MiniMaxText01Config", | |
| "AutoModelForCausalLM": "MiniMaxAI/MiniMax-Text-01--modeling_minimax_text_01.MiniMaxText01ForCausalLM" | |
| }, | |
| "head_dim": 128, | |
| "hidden_act": "silu", | |
| "hidden_size": 6144, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 9216, | |
| "layernorm_full_attention_alpha": 3.5565588200778455, | |
| "layernorm_full_attention_beta": 1.0, | |
| "layernorm_linear_attention_alpha": 3.5565588200778455, | |
| "layernorm_linear_attention_beta": 1.0, | |
| "layernorm_mlp_alpha": 3.5565588200778455, | |
| "layernorm_mlp_beta": 1.0, | |
| "max_position_embeddings": 10240000, | |
| "model_type": "MiniMaxText01", | |
| "num_attention_heads": 64, | |
| "num_experts_per_tok": 2, | |
| "num_hidden_layers": 80, | |
| "num_key_value_heads": 8, | |
| "num_local_experts": 32, | |
| "output_router_logits": false, | |
| "postnorm": true, | |
| "quantization_config": { | |
| "amp": true, | |
| "autoround_version": "0.4.4", | |
| "batch_size": 1, | |
| "bits": 4, | |
| "data_type": "int", | |
| "dataset": "NeelNanda/pile-10k", | |
| "enable_minmax_tuning": true, | |
| "enable_norm_bias_tuning": false, | |
| "enable_quanted_input": true, | |
| "extra_config": { | |
| "model.layers.0.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.1.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.10.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.11.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.12.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.13.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.14.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.15.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.16.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.17.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.18.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.19.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.2.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.20.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.21.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.22.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.23.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.24.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.25.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.26.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.27.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.28.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.29.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.3.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.30.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.31.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.32.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.33.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.34.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.35.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.36.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.37.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.38.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.39.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.4.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.40.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.41.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.42.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.43.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.44.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.45.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.46.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.47.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.48.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.49.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.5.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.50.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.51.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.52.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.53.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.54.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.55.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.56.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.57.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.58.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.59.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.6.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.60.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.61.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.62.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.63.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.64.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.65.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.66.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.67.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.68.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.69.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.7.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.70.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.71.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.72.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.73.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.74.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.75.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.76.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.77.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.78.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.79.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.8.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| }, | |
| "model.layers.9.block_sparse_moe.gate": { | |
| "bits": 16, | |
| "data_type": "bfloat" | |
| } | |
| }, | |
| "gradient_accumulate_steps": 4, | |
| "group_size": 128, | |
| "iters": 200, | |
| "low_gpu_mem_usage": false, | |
| "lr": 0.005, | |
| "minmax_lr": 0.005, | |
| "nsamples": 128, | |
| "quant_method": "auto-round", | |
| "scale_dtype": "torch.float16", | |
| "seqlen": 512, | |
| "sym": true | |
| }, | |
| "rms_norm_eps": 1e-05, | |
| "rope_theta": 10000000, | |
| "rotary_dim": 64, | |
| "router_aux_loss_coef": 0.001, | |
| "router_jitter_noise": 0.0, | |
| "shared_intermediate_size": 0, | |
| "shared_moe_mode": "sigmoid", | |
| "sliding_window": null, | |
| "tie_word_embeddings": false, | |
| "torch_dtype": "bfloat16", | |
| "transformers_version": "4.48.0", | |
| "use_cache": true, | |
| "vocab_size": 200064 | |
| } | |