Instructions to use amd/Step-3.5-Flash-MXFP4 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use amd/Step-3.5-Flash-MXFP4 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="amd/Step-3.5-Flash-MXFP4", trust_remote_code=True) messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("amd/Step-3.5-Flash-MXFP4", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use amd/Step-3.5-Flash-MXFP4 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "amd/Step-3.5-Flash-MXFP4" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "amd/Step-3.5-Flash-MXFP4", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/amd/Step-3.5-Flash-MXFP4
- SGLang
How to use amd/Step-3.5-Flash-MXFP4 with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "amd/Step-3.5-Flash-MXFP4" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "amd/Step-3.5-Flash-MXFP4", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "amd/Step-3.5-Flash-MXFP4" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "amd/Step-3.5-Flash-MXFP4", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use amd/Step-3.5-Flash-MXFP4 with Docker Model Runner:
docker model run hf.co/amd/Step-3.5-Flash-MXFP4
| { | |
| "architectures": [ | |
| "Step3p5ForCausalLM" | |
| ], | |
| "att_impl_type": "GQA", | |
| "attention_other_setting": { | |
| "attention_type": "sliding_attention", | |
| "head_dim": 128, | |
| "num_attention_groups": 8, | |
| "num_attention_heads": 96, | |
| "true_head_dim": 128 | |
| }, | |
| "auto_map": { | |
| "AutoConfig": "configuration_step3p5.Step3p5Config", | |
| "AutoModelForCausalLM": "modeling_step3p5.Step3p5ForCausalLM" | |
| }, | |
| "bos_token_id": 0, | |
| "dtype": "bfloat16", | |
| "eos_token_id": [ | |
| 1, | |
| 2, | |
| 128007 | |
| ], | |
| "head_dim": 128, | |
| "hidden_size": 4096, | |
| "intermediate_size": 11264, | |
| "layer_types": [ | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "full_attention", | |
| "sliding_attention", | |
| "sliding_attention", | |
| "sliding_attention" | |
| ], | |
| "max_position_embeddings": 262144, | |
| "max_seq_len": 262144, | |
| "model_type": "step3p5", | |
| "moe_every_n_layer": 1, | |
| "moe_intermediate_size": 1280, | |
| "moe_layer_offset": 0, | |
| "moe_layers_enum": "3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44", | |
| "moe_num_experts": 288, | |
| "moe_router_activation": "sigmoid", | |
| "moe_router_scaling_factor": 3.0, | |
| "moe_top_k": 8, | |
| "need_fp32_gate": true, | |
| "norm_expert_weight": true, | |
| "num_attention_groups": 8, | |
| "num_attention_heads": 64, | |
| "num_hidden_layers": 45, | |
| "num_nextn_predict_layers": 3, | |
| "partial_rotary_factor": 0.5, | |
| "partial_rotary_factors": [ | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 0.5, | |
| 1.0, | |
| 1.0, | |
| 1.0 | |
| ], | |
| "quantization_config": { | |
| "algo_config": null, | |
| "exclude": [ | |
| "model.embed_tokens", | |
| "model.layers.0.self_attn.q_proj", | |
| "model.layers.0.self_attn.k_proj", | |
| "model.layers.0.self_attn.v_proj", | |
| "model.layers.0.self_attn.o_proj", | |
| "model.layers.0.self_attn.g_proj", | |
| "model.layers.0.mlp.gate_proj", | |
| "model.layers.0.mlp.up_proj", | |
| "model.layers.0.mlp.down_proj", | |
| "model.layers.1.self_attn.q_proj", | |
| "model.layers.1.self_attn.k_proj", | |
| "model.layers.1.self_attn.v_proj", | |
| "model.layers.1.self_attn.o_proj", | |
| "model.layers.1.self_attn.g_proj", | |
| "model.layers.1.mlp.gate_proj", | |
| "model.layers.1.mlp.up_proj", | |
| "model.layers.1.mlp.down_proj", | |
| "model.layers.2.self_attn.q_proj", | |
| "model.layers.2.self_attn.k_proj", | |
| "model.layers.2.self_attn.v_proj", | |
| "model.layers.2.self_attn.o_proj", | |
| "model.layers.2.self_attn.g_proj", | |
| "model.layers.2.mlp.gate_proj", | |
| "model.layers.2.mlp.up_proj", | |
| "model.layers.2.mlp.down_proj", | |
| "model.layers.3.self_attn.q_proj", | |
| "model.layers.3.self_attn.k_proj", | |
| "model.layers.3.self_attn.v_proj", | |
| "model.layers.3.self_attn.o_proj", | |
| "model.layers.3.self_attn.g_proj", | |
| "model.layers.3.moe.gate", | |
| "model.layers.3.share_expert.gate_proj", | |
| "model.layers.3.share_expert.up_proj", | |
| "model.layers.3.share_expert.down_proj", | |
| "model.layers.4.self_attn.q_proj", | |
| "model.layers.4.self_attn.k_proj", | |
| "model.layers.4.self_attn.v_proj", | |
| "model.layers.4.self_attn.o_proj", | |
| "model.layers.4.self_attn.g_proj", | |
| "model.layers.4.moe.gate", | |
| "model.layers.4.share_expert.gate_proj", | |
| "model.layers.4.share_expert.up_proj", | |
| "model.layers.4.share_expert.down_proj", | |
| "model.layers.5.self_attn.q_proj", | |
| "model.layers.5.self_attn.k_proj", | |
| "model.layers.5.self_attn.v_proj", | |
| "model.layers.5.self_attn.o_proj", | |
| "model.layers.5.self_attn.g_proj", | |
| "model.layers.5.moe.gate", | |
| "model.layers.5.share_expert.gate_proj", | |
| "model.layers.5.share_expert.up_proj", | |
| "model.layers.5.share_expert.down_proj", | |
| "model.layers.6.self_attn.q_proj", | |
| "model.layers.6.self_attn.k_proj", | |
| "model.layers.6.self_attn.v_proj", | |
| "model.layers.6.self_attn.o_proj", | |
| "model.layers.6.self_attn.g_proj", | |
| "model.layers.6.moe.gate", | |
| "model.layers.6.share_expert.gate_proj", | |
| "model.layers.6.share_expert.up_proj", | |
| "model.layers.6.share_expert.down_proj", | |
| "model.layers.7.self_attn.q_proj", | |
| "model.layers.7.self_attn.k_proj", | |
| "model.layers.7.self_attn.v_proj", | |
| "model.layers.7.self_attn.o_proj", | |
| "model.layers.7.self_attn.g_proj", | |
| "model.layers.7.moe.gate", | |
| "model.layers.7.share_expert.gate_proj", | |
| "model.layers.7.share_expert.up_proj", | |
| "model.layers.7.share_expert.down_proj", | |
| "model.layers.8.self_attn.q_proj", | |
| "model.layers.8.self_attn.k_proj", | |
| "model.layers.8.self_attn.v_proj", | |
| "model.layers.8.self_attn.o_proj", | |
| "model.layers.8.self_attn.g_proj", | |
| "model.layers.8.moe.gate", | |
| "model.layers.8.share_expert.gate_proj", | |
| "model.layers.8.share_expert.up_proj", | |
| "model.layers.8.share_expert.down_proj", | |
| "model.layers.9.self_attn.q_proj", | |
| "model.layers.9.self_attn.k_proj", | |
| "model.layers.9.self_attn.v_proj", | |
| "model.layers.9.self_attn.o_proj", | |
| "model.layers.9.self_attn.g_proj", | |
| "model.layers.9.moe.gate", | |
| "model.layers.9.share_expert.gate_proj", | |
| "model.layers.9.share_expert.up_proj", | |
| "model.layers.9.share_expert.down_proj", | |
| "model.layers.10.self_attn.q_proj", | |
| "model.layers.10.self_attn.k_proj", | |
| "model.layers.10.self_attn.v_proj", | |
| "model.layers.10.self_attn.o_proj", | |
| "model.layers.10.self_attn.g_proj", | |
| "model.layers.10.moe.gate", | |
| "model.layers.10.share_expert.gate_proj", | |
| "model.layers.10.share_expert.up_proj", | |
| "model.layers.10.share_expert.down_proj", | |
| "model.layers.11.self_attn.q_proj", | |
| "model.layers.11.self_attn.k_proj", | |
| "model.layers.11.self_attn.v_proj", | |
| "model.layers.11.self_attn.o_proj", | |
| "model.layers.11.self_attn.g_proj", | |
| "model.layers.11.moe.gate", | |
| "model.layers.11.share_expert.gate_proj", | |
| "model.layers.11.share_expert.up_proj", | |
| "model.layers.11.share_expert.down_proj", | |
| "model.layers.12.self_attn.q_proj", | |
| "model.layers.12.self_attn.k_proj", | |
| "model.layers.12.self_attn.v_proj", | |
| "model.layers.12.self_attn.o_proj", | |
| "model.layers.12.self_attn.g_proj", | |
| "model.layers.12.moe.gate", | |
| "model.layers.12.share_expert.gate_proj", | |
| "model.layers.12.share_expert.up_proj", | |
| "model.layers.12.share_expert.down_proj", | |
| "model.layers.13.self_attn.q_proj", | |
| "model.layers.13.self_attn.k_proj", | |
| "model.layers.13.self_attn.v_proj", | |
| "model.layers.13.self_attn.o_proj", | |
| "model.layers.13.self_attn.g_proj", | |
| "model.layers.13.moe.gate", | |
| "model.layers.13.share_expert.gate_proj", | |
| "model.layers.13.share_expert.up_proj", | |
| "model.layers.13.share_expert.down_proj", | |
| "model.layers.14.self_attn.q_proj", | |
| "model.layers.14.self_attn.k_proj", | |
| "model.layers.14.self_attn.v_proj", | |
| "model.layers.14.self_attn.o_proj", | |
| "model.layers.14.self_attn.g_proj", | |
| "model.layers.14.moe.gate", | |
| "model.layers.14.share_expert.gate_proj", | |
| "model.layers.14.share_expert.up_proj", | |
| "model.layers.14.share_expert.down_proj", | |
| "model.layers.15.self_attn.q_proj", | |
| "model.layers.15.self_attn.k_proj", | |
| "model.layers.15.self_attn.v_proj", | |
| "model.layers.15.self_attn.o_proj", | |
| "model.layers.15.self_attn.g_proj", | |
| "model.layers.15.moe.gate", | |
| "model.layers.15.share_expert.gate_proj", | |
| "model.layers.15.share_expert.up_proj", | |
| "model.layers.15.share_expert.down_proj", | |
| "model.layers.16.self_attn.q_proj", | |
| "model.layers.16.self_attn.k_proj", | |
| "model.layers.16.self_attn.v_proj", | |
| "model.layers.16.self_attn.o_proj", | |
| "model.layers.16.self_attn.g_proj", | |
| "model.layers.16.moe.gate", | |
| "model.layers.16.share_expert.gate_proj", | |
| "model.layers.16.share_expert.up_proj", | |
| "model.layers.16.share_expert.down_proj", | |
| "model.layers.17.self_attn.q_proj", | |
| "model.layers.17.self_attn.k_proj", | |
| "model.layers.17.self_attn.v_proj", | |
| "model.layers.17.self_attn.o_proj", | |
| "model.layers.17.self_attn.g_proj", | |
| "model.layers.17.moe.gate", | |
| "model.layers.17.share_expert.gate_proj", | |
| "model.layers.17.share_expert.up_proj", | |
| "model.layers.17.share_expert.down_proj", | |
| "model.layers.18.self_attn.q_proj", | |
| "model.layers.18.self_attn.k_proj", | |
| "model.layers.18.self_attn.v_proj", | |
| "model.layers.18.self_attn.o_proj", | |
| "model.layers.18.self_attn.g_proj", | |
| "model.layers.18.moe.gate", | |
| "model.layers.18.share_expert.gate_proj", | |
| "model.layers.18.share_expert.up_proj", | |
| "model.layers.18.share_expert.down_proj", | |
| "model.layers.19.self_attn.q_proj", | |
| "model.layers.19.self_attn.k_proj", | |
| "model.layers.19.self_attn.v_proj", | |
| "model.layers.19.self_attn.o_proj", | |
| "model.layers.19.self_attn.g_proj", | |
| "model.layers.19.moe.gate", | |
| "model.layers.19.share_expert.gate_proj", | |
| "model.layers.19.share_expert.up_proj", | |
| "model.layers.19.share_expert.down_proj", | |
| "model.layers.20.self_attn.q_proj", | |
| "model.layers.20.self_attn.k_proj", | |
| "model.layers.20.self_attn.v_proj", | |
| "model.layers.20.self_attn.o_proj", | |
| "model.layers.20.self_attn.g_proj", | |
| "model.layers.20.moe.gate", | |
| "model.layers.20.share_expert.gate_proj", | |
| "model.layers.20.share_expert.up_proj", | |
| "model.layers.20.share_expert.down_proj", | |
| "model.layers.21.self_attn.q_proj", | |
| "model.layers.21.self_attn.k_proj", | |
| "model.layers.21.self_attn.v_proj", | |
| "model.layers.21.self_attn.o_proj", | |
| "model.layers.21.self_attn.g_proj", | |
| "model.layers.21.moe.gate", | |
| "model.layers.21.share_expert.gate_proj", | |
| "model.layers.21.share_expert.up_proj", | |
| "model.layers.21.share_expert.down_proj", | |
| "model.layers.22.self_attn.q_proj", | |
| "model.layers.22.self_attn.k_proj", | |
| "model.layers.22.self_attn.v_proj", | |
| "model.layers.22.self_attn.o_proj", | |
| "model.layers.22.self_attn.g_proj", | |
| "model.layers.22.moe.gate", | |
| "model.layers.22.share_expert.gate_proj", | |
| "model.layers.22.share_expert.up_proj", | |
| "model.layers.22.share_expert.down_proj", | |
| "model.layers.23.self_attn.q_proj", | |
| "model.layers.23.self_attn.k_proj", | |
| "model.layers.23.self_attn.v_proj", | |
| "model.layers.23.self_attn.o_proj", | |
| "model.layers.23.self_attn.g_proj", | |
| "model.layers.23.moe.gate", | |
| "model.layers.23.share_expert.gate_proj", | |
| "model.layers.23.share_expert.up_proj", | |
| "model.layers.23.share_expert.down_proj", | |
| "model.layers.24.self_attn.q_proj", | |
| "model.layers.24.self_attn.k_proj", | |
| "model.layers.24.self_attn.v_proj", | |
| "model.layers.24.self_attn.o_proj", | |
| "model.layers.24.self_attn.g_proj", | |
| "model.layers.24.moe.gate", | |
| "model.layers.24.share_expert.gate_proj", | |
| "model.layers.24.share_expert.up_proj", | |
| "model.layers.24.share_expert.down_proj", | |
| "model.layers.25.self_attn.q_proj", | |
| "model.layers.25.self_attn.k_proj", | |
| "model.layers.25.self_attn.v_proj", | |
| "model.layers.25.self_attn.o_proj", | |
| "model.layers.25.self_attn.g_proj", | |
| "model.layers.25.moe.gate", | |
| "model.layers.25.share_expert.gate_proj", | |
| "model.layers.25.share_expert.up_proj", | |
| "model.layers.25.share_expert.down_proj", | |
| "model.layers.26.self_attn.q_proj", | |
| "model.layers.26.self_attn.k_proj", | |
| "model.layers.26.self_attn.v_proj", | |
| "model.layers.26.self_attn.o_proj", | |
| "model.layers.26.self_attn.g_proj", | |
| "model.layers.26.moe.gate", | |
| "model.layers.26.share_expert.gate_proj", | |
| "model.layers.26.share_expert.up_proj", | |
| "model.layers.26.share_expert.down_proj", | |
| "model.layers.27.self_attn.q_proj", | |
| "model.layers.27.self_attn.k_proj", | |
| "model.layers.27.self_attn.v_proj", | |
| "model.layers.27.self_attn.o_proj", | |
| "model.layers.27.self_attn.g_proj", | |
| "model.layers.27.moe.gate", | |
| "model.layers.27.share_expert.gate_proj", | |
| "model.layers.27.share_expert.up_proj", | |
| "model.layers.27.share_expert.down_proj", | |
| "model.layers.28.self_attn.q_proj", | |
| "model.layers.28.self_attn.k_proj", | |
| "model.layers.28.self_attn.v_proj", | |
| "model.layers.28.self_attn.o_proj", | |
| "model.layers.28.self_attn.g_proj", | |
| "model.layers.28.moe.gate", | |
| "model.layers.28.share_expert.gate_proj", | |
| "model.layers.28.share_expert.up_proj", | |
| "model.layers.28.share_expert.down_proj", | |
| "model.layers.29.self_attn.q_proj", | |
| "model.layers.29.self_attn.k_proj", | |
| "model.layers.29.self_attn.v_proj", | |
| "model.layers.29.self_attn.o_proj", | |
| "model.layers.29.self_attn.g_proj", | |
| "model.layers.29.moe.gate", | |
| "model.layers.29.share_expert.gate_proj", | |
| "model.layers.29.share_expert.up_proj", | |
| "model.layers.29.share_expert.down_proj", | |
| "model.layers.30.self_attn.q_proj", | |
| "model.layers.30.self_attn.k_proj", | |
| "model.layers.30.self_attn.v_proj", | |
| "model.layers.30.self_attn.o_proj", | |
| "model.layers.30.self_attn.g_proj", | |
| "model.layers.30.moe.gate", | |
| "model.layers.30.share_expert.gate_proj", | |
| "model.layers.30.share_expert.up_proj", | |
| "model.layers.30.share_expert.down_proj", | |
| "model.layers.31.self_attn.q_proj", | |
| "model.layers.31.self_attn.k_proj", | |
| "model.layers.31.self_attn.v_proj", | |
| "model.layers.31.self_attn.o_proj", | |
| "model.layers.31.self_attn.g_proj", | |
| "model.layers.31.moe.gate", | |
| "model.layers.31.share_expert.gate_proj", | |
| "model.layers.31.share_expert.up_proj", | |
| "model.layers.31.share_expert.down_proj", | |
| "model.layers.32.self_attn.q_proj", | |
| "model.layers.32.self_attn.k_proj", | |
| "model.layers.32.self_attn.v_proj", | |
| "model.layers.32.self_attn.o_proj", | |
| "model.layers.32.self_attn.g_proj", | |
| "model.layers.32.moe.gate", | |
| "model.layers.32.share_expert.gate_proj", | |
| "model.layers.32.share_expert.up_proj", | |
| "model.layers.32.share_expert.down_proj", | |
| "model.layers.33.self_attn.q_proj", | |
| "model.layers.33.self_attn.k_proj", | |
| "model.layers.33.self_attn.v_proj", | |
| "model.layers.33.self_attn.o_proj", | |
| "model.layers.33.self_attn.g_proj", | |
| "model.layers.33.moe.gate", | |
| "model.layers.33.share_expert.gate_proj", | |
| "model.layers.33.share_expert.up_proj", | |
| "model.layers.33.share_expert.down_proj", | |
| "model.layers.34.self_attn.q_proj", | |
| "model.layers.34.self_attn.k_proj", | |
| "model.layers.34.self_attn.v_proj", | |
| "model.layers.34.self_attn.o_proj", | |
| "model.layers.34.self_attn.g_proj", | |
| "model.layers.34.moe.gate", | |
| "model.layers.34.share_expert.gate_proj", | |
| "model.layers.34.share_expert.up_proj", | |
| "model.layers.34.share_expert.down_proj", | |
| "model.layers.35.self_attn.q_proj", | |
| "model.layers.35.self_attn.k_proj", | |
| "model.layers.35.self_attn.v_proj", | |
| "model.layers.35.self_attn.o_proj", | |
| "model.layers.35.self_attn.g_proj", | |
| "model.layers.35.moe.gate", | |
| "model.layers.35.share_expert.gate_proj", | |
| "model.layers.35.share_expert.up_proj", | |
| "model.layers.35.share_expert.down_proj", | |
| "model.layers.36.self_attn.q_proj", | |
| "model.layers.36.self_attn.k_proj", | |
| "model.layers.36.self_attn.v_proj", | |
| "model.layers.36.self_attn.o_proj", | |
| "model.layers.36.self_attn.g_proj", | |
| "model.layers.36.moe.gate", | |
| "model.layers.36.share_expert.gate_proj", | |
| "model.layers.36.share_expert.up_proj", | |
| "model.layers.36.share_expert.down_proj", | |
| "model.layers.37.self_attn.q_proj", | |
| "model.layers.37.self_attn.k_proj", | |
| "model.layers.37.self_attn.v_proj", | |
| "model.layers.37.self_attn.o_proj", | |
| "model.layers.37.self_attn.g_proj", | |
| "model.layers.37.moe.gate", | |
| "model.layers.37.share_expert.gate_proj", | |
| "model.layers.37.share_expert.up_proj", | |
| "model.layers.37.share_expert.down_proj", | |
| "model.layers.38.self_attn.q_proj", | |
| "model.layers.38.self_attn.k_proj", | |
| "model.layers.38.self_attn.v_proj", | |
| "model.layers.38.self_attn.o_proj", | |
| "model.layers.38.self_attn.g_proj", | |
| "model.layers.38.moe.gate", | |
| "model.layers.38.share_expert.gate_proj", | |
| "model.layers.38.share_expert.up_proj", | |
| "model.layers.38.share_expert.down_proj", | |
| "model.layers.39.self_attn.q_proj", | |
| "model.layers.39.self_attn.k_proj", | |
| "model.layers.39.self_attn.v_proj", | |
| "model.layers.39.self_attn.o_proj", | |
| "model.layers.39.self_attn.g_proj", | |
| "model.layers.39.moe.gate", | |
| "model.layers.39.share_expert.gate_proj", | |
| "model.layers.39.share_expert.up_proj", | |
| "model.layers.39.share_expert.down_proj", | |
| "model.layers.40.self_attn.q_proj", | |
| "model.layers.40.self_attn.k_proj", | |
| "model.layers.40.self_attn.v_proj", | |
| "model.layers.40.self_attn.o_proj", | |
| "model.layers.40.self_attn.g_proj", | |
| "model.layers.40.moe.gate", | |
| "model.layers.40.share_expert.gate_proj", | |
| "model.layers.40.share_expert.up_proj", | |
| "model.layers.40.share_expert.down_proj", | |
| "model.layers.41.self_attn.q_proj", | |
| "model.layers.41.self_attn.k_proj", | |
| "model.layers.41.self_attn.v_proj", | |
| "model.layers.41.self_attn.o_proj", | |
| "model.layers.41.self_attn.g_proj", | |
| "model.layers.41.moe.gate", | |
| "model.layers.41.share_expert.gate_proj", | |
| "model.layers.41.share_expert.up_proj", | |
| "model.layers.41.share_expert.down_proj", | |
| "model.layers.42.self_attn.q_proj", | |
| "model.layers.42.self_attn.k_proj", | |
| "model.layers.42.self_attn.v_proj", | |
| "model.layers.42.self_attn.o_proj", | |
| "model.layers.42.self_attn.g_proj", | |
| "model.layers.42.moe.gate", | |
| "model.layers.42.share_expert.gate_proj", | |
| "model.layers.42.share_expert.up_proj", | |
| "model.layers.42.share_expert.down_proj", | |
| "model.layers.43.self_attn.q_proj", | |
| "model.layers.43.self_attn.k_proj", | |
| "model.layers.43.self_attn.v_proj", | |
| "model.layers.43.self_attn.o_proj", | |
| "model.layers.43.self_attn.g_proj", | |
| "model.layers.43.moe.gate", | |
| "model.layers.43.share_expert.gate_proj", | |
| "model.layers.43.share_expert.up_proj", | |
| "model.layers.43.share_expert.down_proj", | |
| "model.layers.44.self_attn.q_proj", | |
| "model.layers.44.self_attn.k_proj", | |
| "model.layers.44.self_attn.v_proj", | |
| "model.layers.44.self_attn.o_proj", | |
| "model.layers.44.self_attn.g_proj", | |
| "model.layers.44.moe.gate", | |
| "model.layers.44.share_expert.gate_proj", | |
| "model.layers.44.share_expert.up_proj", | |
| "model.layers.44.share_expert.down_proj", | |
| "lm_head" | |
| ], | |
| "export": { | |
| "kv_cache_group": [], | |
| "min_kv_scale": 0.0, | |
| "pack_method": "reorder", | |
| "weight_format": "real_quantized", | |
| "weight_merge_groups": null | |
| }, | |
| "global_quant_config": { | |
| "bias": null, | |
| "input_tensors": { | |
| "ch_axis": -1, | |
| "dtype": "fp4", | |
| "group_size": 32, | |
| "is_dynamic": true, | |
| "is_scale_quant": false, | |
| "mx_element_dtype": null, | |
| "observer_cls": "PerBlockMXObserver", | |
| "qscheme": "per_group", | |
| "round_method": "half_even", | |
| "scale_calculation_mode": "even", | |
| "scale_format": "e8m0", | |
| "scale_type": "float", | |
| "symmetric": null | |
| }, | |
| "output_tensors": null, | |
| "target_device": null, | |
| "weight": { | |
| "ch_axis": -1, | |
| "dtype": "fp4", | |
| "group_size": 32, | |
| "is_dynamic": false, | |
| "is_scale_quant": false, | |
| "mx_element_dtype": null, | |
| "observer_cls": "PerBlockMXObserver", | |
| "qscheme": "per_group", | |
| "round_method": "half_even", | |
| "scale_calculation_mode": "even", | |
| "scale_format": "e8m0", | |
| "scale_type": "float", | |
| "symmetric": null | |
| } | |
| }, | |
| "kv_cache_post_rope": false, | |
| "kv_cache_quant_config": {}, | |
| "layer_quant_config": {}, | |
| "layer_type_quant_config": {}, | |
| "quant_method": "quark", | |
| "quant_mode": "eager_mode", | |
| "softmax_quant_spec": null, | |
| "version": "0.12+422c9a6d36" | |
| }, | |
| "rms_norm_eps": 1e-05, | |
| "rope_parameters": { | |
| "factor": 2.0, | |
| "high_freq_factor": 32.0, | |
| "low_freq_factor": 1.0, | |
| "original_max_position_embeddings": 131072, | |
| "rope_type": "llama3" | |
| }, | |
| "rope_scaling": { | |
| "factor": 2.0, | |
| "high_freq_factor": 32.0, | |
| "low_freq_factor": 1.0, | |
| "original_max_position_embeddings": 131072, | |
| "rope_type": "llama3" | |
| }, | |
| "rope_theta": [ | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0, | |
| 5000000.0, | |
| 10000.0, | |
| 10000.0, | |
| 10000.0 | |
| ], | |
| "share_expert_dim": 1280, | |
| "sink": false, | |
| "sliding_window": 512, | |
| "swiglu_limits": [ | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 7, | |
| 7, | |
| 0.0, | |
| 0.0, | |
| 0.0 | |
| ], | |
| "swiglu_limits_shared": [ | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 0.0, | |
| 16, | |
| 0.0, | |
| 0.0, | |
| 0.0 | |
| ], | |
| "transformers_version": "4.57.6", | |
| "use_cache": false, | |
| "use_head_wise_attn_gate": true, | |
| "use_moe": true, | |
| "use_moe_router_bias": true, | |
| "use_qk_norm": true, | |
| "use_rope_layers": [], | |
| "vocab_size": 128896, | |
| "yarn_only_types": [ | |
| "full_attention" | |
| ], | |
| "zero_centered": true | |
| } | |