{ "_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", "backend": "auto_round:gptq", "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": "intel/auto-round", "scale_dtype": "torch.float16", "seqlen": 512, "sym": true, "to_quant_block_names": null }, "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 }