Image-Text-to-Text
MLX
Safetensors
English
Chinese
step3p7
jang
jang-k
stepfun
vision-language
conversational
custom_code
modelopt
Instructions to use OsaurusAI/Step-3.7-Flash-JANG_K with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use OsaurusAI/Step-3.7-Flash-JANG_K with MLX:
# Make sure mlx-vlm is installed # pip install --upgrade mlx-vlm from mlx_vlm import load, generate from mlx_vlm.prompt_utils import apply_chat_template from mlx_vlm.utils import load_config # Load the model model, processor = load("OsaurusAI/Step-3.7-Flash-JANG_K") config = load_config("OsaurusAI/Step-3.7-Flash-JANG_K") # Prepare input image = ["http://images.cocodataset.org/val2017/000000039769.jpg"] prompt = "Describe this image." # Apply chat template formatted_prompt = apply_chat_template( processor, config, prompt, num_images=1 ) # Generate output output = generate(model, processor, formatted_prompt, image) print(output) - Notebooks
- Google Colab
- Kaggle
- Local Apps
- LM Studio
- Pi new
How to use OsaurusAI/Step-3.7-Flash-JANG_K with Pi:
Start the MLX server
# Install MLX LM: uv tool install mlx-lm # Start a local OpenAI-compatible server: mlx_lm.server --model "OsaurusAI/Step-3.7-Flash-JANG_K"
Configure the model in Pi
# Install Pi: npm install -g @mariozechner/pi-coding-agent # Add to ~/.pi/agent/models.json: { "providers": { "mlx-lm": { "baseUrl": "http://localhost:8080/v1", "api": "openai-completions", "apiKey": "none", "models": [ { "id": "OsaurusAI/Step-3.7-Flash-JANG_K" } ] } } }Run Pi
# Start Pi in your project directory: pi
- Hermes Agent new
How to use OsaurusAI/Step-3.7-Flash-JANG_K with Hermes Agent:
Start the MLX server
# Install MLX LM: uv tool install mlx-lm # Start a local OpenAI-compatible server: mlx_lm.server --model "OsaurusAI/Step-3.7-Flash-JANG_K"
Configure Hermes
# Install Hermes: curl -fsSL https://hermes-agent.nousresearch.com/install.sh | bash hermes setup # Point Hermes at the local server: hermes config set model.provider custom hermes config set model.base_url http://127.0.0.1:8080/v1 hermes config set model.default OsaurusAI/Step-3.7-Flash-JANG_K
Run Hermes
hermes
| { | |
| "architectures": [ | |
| "Step3p7ForConditionalGeneration" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_step3p7.Step3p7Config", | |
| "AutoModelForCausalLM": "modeling_step3p7.Step3p7ForConditionalGeneration", | |
| "AutoProcessor": "processing_step3.Step3VLProcessor" | |
| }, | |
| "dtype": "bfloat16", | |
| "hidden_size": 4096, | |
| "im_end_token": "<im_end>", | |
| "im_patch_token": "<im_patch>", | |
| "im_start_token": "<im_start>", | |
| "image_token_id": 128001, | |
| "image_token_len": 169, | |
| "jang_profile": "JANG_K", | |
| "max_position_embeddings": 262144, | |
| "model_file": "step3p7_mlx.py", | |
| "model_type": "step3p7", | |
| "mxtq_bits": { | |
| "attention": 8, | |
| "embedding": 6, | |
| "routed_expert": { | |
| "down_proj": 2, | |
| "gate_proj": 4, | |
| "up_proj": 2 | |
| } | |
| }, | |
| "mxtq_seed": 42, | |
| "patch_token_len": 81, | |
| "projector_bias": false, | |
| "quantization": { | |
| "bits": 2, | |
| "format": "jang", | |
| "group_size": 128, | |
| "lm_head": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.embed_tokens": { | |
| "bits": 6, | |
| "group_size": 128 | |
| }, | |
| "model.layers.0.mlp.down_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.0.mlp.gate_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.0.mlp.up_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.0.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.0.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.0.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.0.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.0.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.1.mlp.down_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.1.mlp.gate_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.1.mlp.up_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.1.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.1.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.1.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.1.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.1.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.10.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.10.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.11.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.11.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.12.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.12.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.13.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.13.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.14.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.14.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.15.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.15.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.16.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.16.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.17.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.17.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.18.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.18.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.19.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.19.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.2.mlp.down_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.2.mlp.gate_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.2.mlp.up_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.2.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.2.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.2.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.2.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.2.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.20.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.20.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.21.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.21.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.22.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.22.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.23.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.23.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.24.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.24.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.25.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.25.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.26.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.26.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.27.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.27.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.28.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.28.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.29.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.29.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.3.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.3.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.30.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.30.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.31.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.31.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.32.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.32.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.33.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.33.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.34.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.34.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.35.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.35.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.36.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.36.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.37.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.37.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.38.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.38.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.39.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.39.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.4.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.4.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.40.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.40.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.41.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.41.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.42.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.42.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.43.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.43.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.44.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.44.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.5.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.5.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.6.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.6.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.7.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.7.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.8.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.8.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.mlp.gate.gate": { | |
| "bits": 8, | |
| "group_size": 64 | |
| }, | |
| "model.layers.9.mlp.share_expert.down_proj": { | |
| "bits": 3, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.mlp.share_expert.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.mlp.share_expert.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.mlp.switch_mlp.down_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.mlp.switch_mlp.gate_proj": { | |
| "bits": 4, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.mlp.switch_mlp.up_proj": { | |
| "bits": 2, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.self_attn.g_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.self_attn.k_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.self_attn.o_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.self_attn.q_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "model.layers.9.self_attn.v_proj": { | |
| "bits": 8, | |
| "group_size": 128 | |
| }, | |
| "profile": "JANG_K" | |
| }, | |
| "quantization_config": { | |
| "config_groups": { | |
| "group_0": { | |
| "input_activations": { | |
| "dynamic": false, | |
| "group_size": 16, | |
| "num_bits": 4, | |
| "type": "float" | |
| }, | |
| "targets": [ | |
| "Linear" | |
| ], | |
| "weights": { | |
| "dynamic": false, | |
| "group_size": 16, | |
| "num_bits": 4, | |
| "type": "float" | |
| } | |
| } | |
| }, | |
| "ignore": [ | |
| "lm_head", | |
| "model.language_model.layers.0*", | |
| "model.language_model.layers.1.*", | |
| "model.language_model.layers.10.moe.gate", | |
| "model.language_model.layers.10.self_attn*", | |
| "model.language_model.layers.10.share_expert*", | |
| "model.language_model.layers.11.moe.gate", | |
| "model.language_model.layers.11.self_attn*", | |
| "model.language_model.layers.11.share_expert*", | |
| "model.language_model.layers.12.moe.gate", | |
| "model.language_model.layers.12.self_attn*", | |
| "model.language_model.layers.12.share_expert*", | |
| "model.language_model.layers.13.moe.gate", | |
| "model.language_model.layers.13.self_attn*", | |
| "model.language_model.layers.13.share_expert*", | |
| "model.language_model.layers.14.moe.gate", | |
| "model.language_model.layers.14.self_attn*", | |
| "model.language_model.layers.14.share_expert*", | |
| "model.language_model.layers.15.moe.gate", | |
| "model.language_model.layers.15.self_attn*", | |
| "model.language_model.layers.15.share_expert*", | |
| "model.language_model.layers.16.moe.gate", | |
| "model.language_model.layers.16.self_attn*", | |
| "model.language_model.layers.16.share_expert*", | |
| "model.language_model.layers.17.moe.gate", | |
| "model.language_model.layers.17.self_attn*", | |
| "model.language_model.layers.17.share_expert*", | |
| "model.language_model.layers.18.moe.gate", | |
| "model.language_model.layers.18.self_attn*", | |
| "model.language_model.layers.18.share_expert*", | |
| "model.language_model.layers.19.moe.gate", | |
| "model.language_model.layers.19.self_attn*", | |
| "model.language_model.layers.19.share_expert*", | |
| "model.language_model.layers.2.*", | |
| "model.language_model.layers.20.moe.gate", | |
| "model.language_model.layers.20.self_attn*", | |
| "model.language_model.layers.20.share_expert*", | |
| "model.language_model.layers.21.moe.gate", | |
| "model.language_model.layers.21.self_attn*", | |
| "model.language_model.layers.21.share_expert*", | |
| "model.language_model.layers.22.moe.gate", | |
| "model.language_model.layers.22.self_attn*", | |
| "model.language_model.layers.22.share_expert*", | |
| "model.language_model.layers.23.moe.gate", | |
| "model.language_model.layers.23.self_attn*", | |
| "model.language_model.layers.23.share_expert*", | |
| "model.language_model.layers.24.moe.gate", | |
| "model.language_model.layers.24.self_attn*", | |
| "model.language_model.layers.24.share_expert*", | |
| "model.language_model.layers.25.moe.gate", | |
| "model.language_model.layers.25.self_attn*", | |
| "model.language_model.layers.25.share_expert*", | |
| "model.language_model.layers.26.moe.gate", | |
| "model.language_model.layers.26.self_attn*", | |
| "model.language_model.layers.26.share_expert*", | |
| "model.language_model.layers.27.moe.gate", | |
| "model.language_model.layers.27.self_attn*", | |
| "model.language_model.layers.27.share_expert*", | |
| "model.language_model.layers.28.moe.gate", | |
| "model.language_model.layers.28.self_attn*", | |
| "model.language_model.layers.28.share_expert*", | |
| "model.language_model.layers.29.moe.gate", | |
| "model.language_model.layers.29.self_attn*", | |
| "model.language_model.layers.29.share_expert*", | |
| "model.language_model.layers.3.moe.gate", | |
| "model.language_model.layers.3.self_attn*", | |
| "model.language_model.layers.3.share_expert*", | |
| "model.language_model.layers.30.moe.gate", | |
| "model.language_model.layers.30.self_attn*", | |
| "model.language_model.layers.30.share_expert*", | |
| "model.language_model.layers.31.moe.gate", | |
| "model.language_model.layers.31.self_attn*", | |
| "model.language_model.layers.31.share_expert*", | |
| "model.language_model.layers.32.moe.gate", | |
| "model.language_model.layers.32.self_attn*", | |
| "model.language_model.layers.32.share_expert*", | |
| "model.language_model.layers.33.moe.gate", | |
| "model.language_model.layers.33.self_attn*", | |
| "model.language_model.layers.33.share_expert*", | |
| "model.language_model.layers.34.moe.gate", | |
| "model.language_model.layers.34.self_attn*", | |
| "model.language_model.layers.34.share_expert*", | |
| "model.language_model.layers.35.moe.gate", | |
| "model.language_model.layers.35.self_attn*", | |
| "model.language_model.layers.35.share_expert*", | |
| "model.language_model.layers.36.moe.gate", | |
| "model.language_model.layers.36.self_attn*", | |
| "model.language_model.layers.36.share_expert*", | |
| "model.language_model.layers.37.moe.gate", | |
| "model.language_model.layers.37.self_attn*", | |
| "model.language_model.layers.37.share_expert*", | |
| "model.language_model.layers.38.moe.gate", | |
| "model.language_model.layers.38.self_attn*", | |
| "model.language_model.layers.38.share_expert*", | |
| "model.language_model.layers.39.moe.gate", | |
| "model.language_model.layers.39.self_attn*", | |
| "model.language_model.layers.39.share_expert*", | |
| "model.language_model.layers.4.moe.gate", | |
| "model.language_model.layers.4.self_attn*", | |
| "model.language_model.layers.4.share_expert*", | |
| "model.language_model.layers.40.moe.gate", | |
| "model.language_model.layers.40.self_attn*", | |
| "model.language_model.layers.40.share_expert*", | |
| "model.language_model.layers.41.moe.gate", | |
| "model.language_model.layers.41.self_attn*", | |
| "model.language_model.layers.41.share_expert*", | |
| "model.language_model.layers.42.moe.gate", | |
| "model.language_model.layers.42.self_attn*", | |
| "model.language_model.layers.42.share_expert*", | |
| "model.language_model.layers.43.moe.gate", | |
| "model.language_model.layers.43.self_attn*", | |
| "model.language_model.layers.43.share_expert*", | |
| "model.language_model.layers.44.moe.gate", | |
| "model.language_model.layers.44.self_attn*", | |
| "model.language_model.layers.44.share_expert*", | |
| "model.language_model.layers.5.moe.gate", | |
| "model.language_model.layers.5.self_attn*", | |
| "model.language_model.layers.5.share_expert*", | |
| "model.language_model.layers.6.moe.gate", | |
| "model.language_model.layers.6.self_attn*", | |
| "model.language_model.layers.6.share_expert*", | |
| "model.language_model.layers.7.moe.gate", | |
| "model.language_model.layers.7.self_attn*", | |
| "model.language_model.layers.7.share_expert*", | |
| "model.language_model.layers.8.moe.gate", | |
| "model.language_model.layers.8.self_attn*", | |
| "model.language_model.layers.8.share_expert*", | |
| "model.language_model.layers.9.moe.gate", | |
| "model.language_model.layers.9.self_attn*", | |
| "model.language_model.layers.9.share_expert*", | |
| "model.vision_model*", | |
| "model.vit_large_projector" | |
| ], | |
| "kv_cache_scheme": { | |
| "dynamic": false, | |
| "num_bits": 8, | |
| "type": "float" | |
| }, | |
| "producer": { | |
| "name": "modelopt", | |
| "version": "0.45.0.dev37+g3ad4f4f09.d20260524" | |
| }, | |
| "quant_algo": "NVFP4", | |
| "quant_method": "modelopt" | |
| }, | |
| "routed_expert_bits": { | |
| "down_proj": 2, | |
| "gate_proj": 4, | |
| "up_proj": 2 | |
| }, | |
| "text_config": { | |
| "architectures": [ | |
| "Step3p5ForCausalLM" | |
| ], | |
| "att_impl_type": "GQA", | |
| "attention_dropout": 0.0, | |
| "attention_other_setting": { | |
| "attention_type": "sliding_attention", | |
| "head_dim": 128, | |
| "num_attention_groups": 8, | |
| "num_attention_heads": 96, | |
| "true_head_dim": 128 | |
| }, | |
| "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" | |
| ], | |
| "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 | |
| ], | |
| "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_theta": 500000.0, | |
| "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 | |
| ], | |
| "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, | |
| 16, | |
| 16 | |
| ], | |
| "torch_dtype": "bfloat16", | |
| "use_cache": true, | |
| "use_head_wise_attn_gate": true, | |
| "use_mfa": false, | |
| "use_moe": true, | |
| "use_moe_router_bias": true, | |
| "use_qk_norm": false, | |
| "use_rope_layers": [], | |
| "vocab_size": 128896, | |
| "yarn_only_types": [ | |
| "full_attention" | |
| ] | |
| }, | |
| "transformers_version": "4.56.2", | |
| "understand_projector_stride": 2, | |
| "use_cache": true, | |
| "use_im_start_end": "true", | |
| "vision_config": { | |
| "heads": 16, | |
| "hidden_act": "quick_gelu", | |
| "image_size": 728, | |
| "layer_norm_eps": 1e-05, | |
| "layers": 47, | |
| "ls_init_value": 0.1, | |
| "mlp_ratio": 5.833333333333333, | |
| "model_type": "perception_encoder", | |
| "num_channels": 3, | |
| "output_dim": null, | |
| "patch_size": 14, | |
| "pool_type": "none", | |
| "ues_cls_token": false, | |
| "use_abs_posemb": true, | |
| "use_cls_token": false, | |
| "use_ln_post": false, | |
| "use_ln_pre": true, | |
| "use_rope2d": true, | |
| "width": 1536 | |
| }, | |
| "vision_select_layer": -1 | |
| } |