Upload folder using huggingface_hub

README.md

@@ -1,3 +1,77 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+language:
+- en
+tags:
+- moe
+- olmo
+- flexolmo
+co2_eq_emissions: 1
+library_name: transformers
+---
+
+<img alt="FlexOlmo Logo." src="FlexOlmo_Logo.png" width="500px" style="display: block; margin-left: auto; margin-right: auto; margin-top: 50px"> FlexOlmo is a new kind of LM that unlocks a new paradigm of data collaboration. With FlexOlmo, data owners can contribute to the development of open language models without giving up control of their data. There is no need to share raw data directly, and data contributors can decide when their data is active in the model, deactivate it at any time, and receive attributions whenever it's used for inference.
+
+
+# Model Summary
+> FlexOlmo-7x7B-1T (without router training) is a Mixture-of-Experts with 33B total parameters, combining independently trained experts on public-mix, news, math, code, academic texts, creative writing, and Reddit data. The public-mix expert is trained on 1T tokens of public data while the other experts are branched from the public-mix expert and trained on 50B tokens of their respective data.
+
+This information and more can also be found:
+- **Paper**: https://allenai.org/papers/flexolmo
+- **Code**: https://github.com/allenai/FlexOlmo
+- **Blog**: https://allenai.org/blog/flexolmo
+- **Data and corresponding models**:
+| Corpus            | Public | Math | News           | Academic          | Code           | Creative Writing | Reddit         |
+|------------------|----------------|----------------|----------------|----------------|----------------|------------------|----------------|
+| Model            |  [Flex-public-7B-1T](https://huggingface.co/allenai/Flex-public-7B-1T) |  [Flex-math-2x7B-1T](https://huggingface.co/allenai/Flex-math-2x7B-1T) | [Flex-news-2x7B-1T](https://huggingface.co/allenai/Flex-news-2x7B-1T) | [Flex-pes2o-2x7B-1T](https://huggingface.co/allenai/Flex-pes2o-2x7B-1T) | [Flex-code-2x7B-1T](https://huggingface.co/allenai/Flex-code-2x7B-1T) |  [Flex-creative-2x7B-1T](https://huggingface.co/allenai/Flex-creative-2x7B-1T) | [Flex-reddit-2x7B-1T](https://huggingface.co/allenai/Flex-reddit-2x7B-1T) |
+
+
+# Use
+
+Install `transformers` **from [this source](https://github.com/swj0419/transformers_flexolmo)** and run:
+```python
+from transformers import Olmoe2ForCausalLM, AutoTokenizer
+import torch
+
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+
+MODEL_NAME = "allenai/FlexOlmo-7x7B-1T"
+model = Olmoe2ForCausalLM.from_pretrained(MODEL_NAME).to(DEVICE)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+inputs = tokenizer("Bitcoin is", return_tensors="pt")
+inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
+out = model.generate(**inputs, max_length=64)
+print(tokenizer.decode(out[0]))
+```
+
+# Evaluation Snapshot
+| **Model** | **MC9** | **Gen5** | **MMLU** | **MMLU Pro** | **AGIEval** | **BBH** | **Math2** | **NewsG** | **PoemG** | **SciRIFF5** | **Code4** | **Avg.** |
+|----------|--------|----------|----------|--------------|-------------|---------|-----------|-----------|-----------|--------------|-----------|----------|
+| Prev. Public model | 68.7 | 58.8 | 55.9 | 26.2 | 39.9 | 35.7 | 8.2 | 76.0 | 47.8 | 48.1 | 1.1 | 42.4 |
+| **Individual** |
+| Math | 62.5 | 44.3 | 50.6 | 24.1 | 42.0 | 45.6 | **53.1** | 42.6 | 28.0 | 50.7 | 15.8 | 41.8 |
+| Code| 40.5 | 39.4 | 29.5 | 14.5 | 27.4 | 38.1 | 6.0 | 45.1 | 28.2 | 48.0 | 21.0 | 30.7 |
+| News | 46.5 | 48.6 | 36.4 | 15.2 | 25.7 | 30.9 | 2.5 | 77.7 | 26.9 | 47.0 | 0.0 | 32.5 |
+| Creative Writing | 42.7 | 43.9 | 31.5 | 11.6 | 23.3 | 27.6 | 1.7 | 56.9 | **67.5** | 42.4 | 0.0 | 31.7 |
+| Academic | 41.0 | 45.2 | 33.8 | 14.8 | 24.1 | 32.4 | 6.5 | 51.8 | 23.0 | 52.0 | 0.0 | 29.5 |
+| Reddit | 64.7 | 36.5 | 56.1 | 25.5 | 35.5 | 19.7 | 2.5 | 54.1 | 8.6 | 32.7 | 1.7 | 30.7 |
+| **Combined** |
+| BTM (top-2) | 68.7 | 57.7 | 59.4 | 28.3 | 43.2 | 44.3 | 23.1 | 73.6 | 54.4 | 46.3 | **24.0** | 47.6 |
+| 🔥 **FlexOlmo-7x7B-1T** | **70.4** | **60.1** | **60.2** | **30.5** | 44.8 | 46.8 | 47.9 | **78.3** | 66.2 | 53.8 | 14.6 | 52.0 |
+| **FlexOlmo-7x7B-1T-RT** | 70.3 | 60.0 | **60.2** | 30.3 | **45.2** | **47.2** | 47.7 | 77.2 | **67.6** | **53.9** | 13.3 | **52.2** |
+
+* The evaluation of the individual model refers to the dense model, not the 2x7B MoE model.
+
+
+# Citation
+```bibtex
+@misc{flexolmo,
+      title={FlexOlmo: Open Language Models for Flexible Data Use}, 
+      author={Weijia Shi and Akshita Bhagia and Kevin Farhat and Niklas Muennighoff and Jacob Morrison and Evan Pete Walsh and Dustin Schwenk and Shayne Longpre and Jake Poznanski and Allyson Ettinger and Daogao Liu and Margaret Li and Mike Lewis and Wen-tau Yih and Dirk Groeneveld and Luca Soldaini and Kyle Lo and Noah A. Smith and Luke Zettlemoyer and Pang Wei Koh and Hannaneh Hajishirzi and Ali Farhadi and Sewon Min},
+      year={2025},
+      eprint={2507.00000},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://allenai.org/papers/flexolmo}, 
+}
+```

config.json

@@ -0,0 +1,32 @@
+{
+  "architectures": [
+    "Olmoe2ForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "clip_qkv": null,
+  "eos_token_id": 100257,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 11008,
+  "max_position_embeddings": 4096,
+  "model_type": "olmoe2",
+  "norm_topk_prob": false,
+  "num_attention_heads": 32,
+  "num_experts": 2,
+  "num_experts_per_tok": 2,
+  "num_hidden_layers": 32,
+  "num_key_value_heads": 32,
+  "output_router_logits": false,
+  "pad_token_id": 100277,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 500000,
+  "router_aux_loss_coef": 0.01,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.50.0.dev0",
+  "use_cache": true,
+  "vocab_size": 100352
+}

generation_config.json

@@ -0,0 +1,4 @@
+{
+  "_from_model_config": true,
+  "transformers_version": "4.50.0.dev0"
+}

model-00001-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7cd9ff2e78ea0791c807b280cb3729eb8fcf7b47216f8a745da34f45fd3e72cb
+size 4974711408

model-00002-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc8411373d2305988fad6ba556477354503df5a16d7db284ea989ef0b9cb4fbd
+size 4974778640

model-00003-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ad96a9b1850c2e26c0db6e433f018a3c2055f5b7a2a61413249a6334b1c1b31
+size 4840593536

model-00004-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83ddfd565fa64859c20572f264b4105b6748842a9710911e4a123491a3040e97
+size 4861598344

model-00005-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1997b28011840c0153f1ad5ef8d3c3448290a13c653435a302e95472f829af1a
+size 4953806944

model-00006-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e57f842036cd31094cc6e80e4c3ea7aa501265a4c34965f14fae59bf67ccc69
+size 4861565336

model-00007-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51f0f98bc42a9d022b45dd0af8b675b10bd477a9f6d66d9c61ff4b94bac2af35
+size 4861598344

model-00008-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01c2ada616e898d39c2494178d4e1dbd52a4b64ffe0b2fbeae62c25435e1ca75
+size 4953773936

model-00009-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3c808603de38189b3e9e039ed00d69a1453eef14aa08f9710594c86246cc32fb
+size 4861598344

model-00010-of-00010.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c5a8f308ebfc5091793525e60636f890428949874b12e3251b0cbacae4a6c19
+size 2365637568

model.safetensors.index.json

@@ -0,0 +1,490 @@
+{
+  "metadata": {
+    "total_size": 46509604864
+  },
+  "weight_map": {
+    "lm_head.weight": "model-00010-of-00010.safetensors",
+    "model.embed_tokens.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.mlp.experts.0.down_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.mlp.experts.0.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.mlp.experts.0.up_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.mlp.experts.1.down_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.mlp.experts.1.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.mlp.experts.1.up_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.mlp.gate.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.post_feedforward_layernorm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.self_attn.k_norm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.self_attn.q_norm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.mlp.experts.0.down_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.mlp.experts.0.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.mlp.experts.0.up_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.mlp.experts.1.down_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.mlp.experts.1.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.mlp.experts.1.up_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.mlp.gate.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.post_feedforward_layernorm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.self_attn.k_norm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.self_attn.q_norm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.10.mlp.experts.0.down_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.mlp.experts.0.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.mlp.experts.0.up_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.mlp.experts.1.down_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.mlp.experts.1.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.mlp.experts.1.up_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.mlp.gate.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.post_attention_layernorm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.post_feedforward_layernorm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.self_attn.k_norm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.self_attn.k_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.self_attn.o_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.self_attn.q_norm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.self_attn.q_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.10.self_attn.v_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.mlp.experts.0.down_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.mlp.experts.0.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.mlp.experts.0.up_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.mlp.experts.1.down_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.mlp.experts.1.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.mlp.experts.1.up_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.mlp.gate.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.post_attention_layernorm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.post_feedforward_layernorm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.self_attn.k_norm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.self_attn.k_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.self_attn.o_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.self_attn.q_norm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.self_attn.q_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.11.self_attn.v_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.mlp.experts.0.down_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.mlp.experts.0.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.mlp.experts.0.up_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.mlp.experts.1.down_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.mlp.experts.1.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.mlp.experts.1.up_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.mlp.gate.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.post_attention_layernorm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.post_feedforward_layernorm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.self_attn.k_norm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.self_attn.k_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.self_attn.o_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.self_attn.q_norm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.self_attn.q_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.12.self_attn.v_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.13.mlp.experts.0.down_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.13.mlp.experts.0.gate_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.13.mlp.experts.0.up_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.13.mlp.experts.1.down_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.13.mlp.experts.1.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.13.mlp.experts.1.up_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.13.mlp.gate.weight": "model-00004-of-00010.safetensors",
+    "model.layers.13.post_attention_layernorm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.13.post_feedforward_layernorm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.13.self_attn.k_norm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.13.self_attn.k_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.13.self_attn.o_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.13.self_attn.q_norm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.13.self_attn.q_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.13.self_attn.v_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.14.mlp.experts.0.down_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.mlp.experts.0.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.mlp.experts.0.up_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.mlp.experts.1.down_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.mlp.experts.1.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.mlp.experts.1.up_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.mlp.gate.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.post_attention_layernorm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.post_feedforward_layernorm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.self_attn.k_norm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.self_attn.k_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.self_attn.o_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.self_attn.q_norm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.self_attn.q_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.14.self_attn.v_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.mlp.experts.0.down_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.mlp.experts.0.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.mlp.experts.0.up_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.mlp.experts.1.down_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.mlp.experts.1.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.mlp.experts.1.up_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.mlp.gate.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.post_attention_layernorm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.post_feedforward_layernorm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.self_attn.k_norm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.self_attn.k_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.self_attn.o_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.self_attn.q_norm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.self_attn.q_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.15.self_attn.v_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.mlp.experts.0.down_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.mlp.experts.0.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.mlp.experts.0.up_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.mlp.experts.1.down_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.mlp.experts.1.gate_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.mlp.experts.1.up_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.mlp.gate.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.post_attention_layernorm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.post_feedforward_layernorm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.self_attn.k_norm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.self_attn.k_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.self_attn.o_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.self_attn.q_norm.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.self_attn.q_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.16.self_attn.v_proj.weight": "model-00005-of-00010.safetensors",
+    "model.layers.17.mlp.experts.0.down_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.mlp.experts.0.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.mlp.experts.0.up_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.mlp.experts.1.down_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.mlp.experts.1.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.mlp.experts.1.up_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.mlp.gate.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.post_attention_layernorm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.post_feedforward_layernorm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.self_attn.k_norm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.self_attn.q_norm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.17.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.mlp.experts.0.down_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.mlp.experts.0.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.mlp.experts.0.up_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.mlp.experts.1.down_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.mlp.experts.1.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.mlp.experts.1.up_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.mlp.gate.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.post_attention_layernorm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.post_feedforward_layernorm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.self_attn.k_norm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.self_attn.q_norm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.18.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.mlp.experts.0.down_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.mlp.experts.0.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.mlp.experts.0.up_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.mlp.experts.1.down_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.mlp.experts.1.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.mlp.experts.1.up_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.mlp.gate.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.post_attention_layernorm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.post_feedforward_layernorm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.self_attn.k_norm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.self_attn.q_norm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.19.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.2.mlp.experts.0.down_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.2.mlp.experts.0.gate_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.2.mlp.experts.0.up_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.2.mlp.experts.1.down_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.2.mlp.experts.1.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.2.mlp.experts.1.up_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.2.mlp.gate.weight": "model-00001-of-00010.safetensors",
+    "model.layers.2.post_attention_layernorm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.2.post_feedforward_layernorm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.2.self_attn.k_norm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.2.self_attn.q_norm.weight": "model-00001-of-00010.safetensors",
+    "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00010.safetensors",
+    "model.layers.20.mlp.experts.0.down_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.mlp.experts.0.gate_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.mlp.experts.0.up_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.mlp.experts.1.down_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.20.mlp.experts.1.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.20.mlp.experts.1.up_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.20.mlp.gate.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.post_attention_layernorm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.20.post_feedforward_layernorm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.20.self_attn.k_norm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.self_attn.k_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.self_attn.o_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.self_attn.q_norm.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.self_attn.q_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.20.self_attn.v_proj.weight": "model-00006-of-00010.safetensors",
+    "model.layers.21.mlp.experts.0.down_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.mlp.experts.0.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.mlp.experts.0.up_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.mlp.experts.1.down_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.mlp.experts.1.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.mlp.experts.1.up_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.mlp.gate.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.post_attention_layernorm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.post_feedforward_layernorm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.self_attn.k_norm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.self_attn.k_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.self_attn.o_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.self_attn.q_norm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.self_attn.q_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.21.self_attn.v_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.mlp.experts.0.down_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.mlp.experts.0.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.mlp.experts.0.up_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.mlp.experts.1.down_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.mlp.experts.1.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.mlp.experts.1.up_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.mlp.gate.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.post_attention_layernorm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.post_feedforward_layernorm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.self_attn.k_norm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.self_attn.k_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.self_attn.o_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.self_attn.q_norm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.self_attn.q_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.22.self_attn.v_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.mlp.experts.0.down_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.mlp.experts.0.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.mlp.experts.0.up_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.mlp.experts.1.down_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.mlp.experts.1.gate_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.mlp.experts.1.up_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.mlp.gate.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.post_attention_layernorm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.post_feedforward_layernorm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.self_attn.k_norm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.self_attn.k_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.self_attn.o_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.self_attn.q_norm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.self_attn.q_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.23.self_attn.v_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.24.mlp.experts.0.down_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.24.mlp.experts.0.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.24.mlp.experts.0.up_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.24.mlp.experts.1.down_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.24.mlp.experts.1.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.24.mlp.experts.1.up_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.24.mlp.gate.weight": "model-00007-of-00010.safetensors",
+    "model.layers.24.post_attention_layernorm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.24.post_feedforward_layernorm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.24.self_attn.k_norm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.24.self_attn.k_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.24.self_attn.o_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.24.self_attn.q_norm.weight": "model-00007-of-00010.safetensors",
+    "model.layers.24.self_attn.q_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.24.self_attn.v_proj.weight": "model-00007-of-00010.safetensors",
+    "model.layers.25.mlp.experts.0.down_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.mlp.experts.0.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.mlp.experts.0.up_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.mlp.experts.1.down_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.mlp.experts.1.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.mlp.experts.1.up_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.mlp.gate.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.post_attention_layernorm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.post_feedforward_layernorm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.self_attn.k_norm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.self_attn.k_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.self_attn.o_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.self_attn.q_norm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.self_attn.q_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.25.self_attn.v_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.mlp.experts.0.down_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.mlp.experts.0.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.mlp.experts.0.up_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.mlp.experts.1.down_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.mlp.experts.1.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.mlp.experts.1.up_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.mlp.gate.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.post_attention_layernorm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.post_feedforward_layernorm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.self_attn.k_norm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.self_attn.k_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.self_attn.o_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.self_attn.q_norm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.self_attn.q_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.26.self_attn.v_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.mlp.experts.0.down_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.mlp.experts.0.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.mlp.experts.0.up_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.mlp.experts.1.down_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.27.mlp.experts.1.gate_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.mlp.experts.1.up_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.mlp.gate.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.27.post_feedforward_layernorm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.27.self_attn.k_norm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.self_attn.k_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.self_attn.o_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.self_attn.q_norm.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.self_attn.q_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.27.self_attn.v_proj.weight": "model-00008-of-00010.safetensors",
+    "model.layers.28.mlp.experts.0.down_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.mlp.experts.0.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.mlp.experts.0.up_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.mlp.experts.1.down_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.mlp.experts.1.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.mlp.experts.1.up_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.mlp.gate.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.post_feedforward_layernorm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.self_attn.k_norm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.self_attn.q_norm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.28.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.mlp.experts.0.down_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.mlp.experts.0.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.mlp.experts.0.up_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.mlp.experts.1.down_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.mlp.experts.1.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.mlp.experts.1.up_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.mlp.gate.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.post_feedforward_layernorm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.self_attn.k_norm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.self_attn.q_norm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.29.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.3.mlp.experts.0.down_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.mlp.experts.0.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.mlp.experts.0.up_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.mlp.experts.1.down_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.mlp.experts.1.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.mlp.experts.1.up_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.mlp.gate.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.post_feedforward_layernorm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.self_attn.k_norm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.self_attn.k_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.self_attn.o_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.self_attn.q_norm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.self_attn.q_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.3.self_attn.v_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.30.mlp.experts.0.down_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.mlp.experts.0.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.mlp.experts.0.up_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.mlp.experts.1.down_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.mlp.experts.1.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.mlp.experts.1.up_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.mlp.gate.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.post_attention_layernorm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.post_feedforward_layernorm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.self_attn.k_norm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.self_attn.q_norm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.30.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.mlp.experts.0.down_proj.weight": "model-00010-of-00010.safetensors",
+    "model.layers.31.mlp.experts.0.gate_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.mlp.experts.0.up_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.mlp.experts.1.down_proj.weight": "model-00010-of-00010.safetensors",
+    "model.layers.31.mlp.experts.1.gate_proj.weight": "model-00010-of-00010.safetensors",
+    "model.layers.31.mlp.experts.1.up_proj.weight": "model-00010-of-00010.safetensors",
+    "model.layers.31.mlp.gate.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.post_attention_layernorm.weight": "model-00010-of-00010.safetensors",
+    "model.layers.31.post_feedforward_layernorm.weight": "model-00010-of-00010.safetensors",
+    "model.layers.31.self_attn.k_norm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.self_attn.k_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.self_attn.o_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.self_attn.q_norm.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.self_attn.q_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.31.self_attn.v_proj.weight": "model-00009-of-00010.safetensors",
+    "model.layers.4.mlp.experts.0.down_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.mlp.experts.0.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.mlp.experts.0.up_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.mlp.experts.1.down_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.mlp.experts.1.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.mlp.experts.1.up_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.mlp.gate.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.post_feedforward_layernorm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.self_attn.k_norm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.self_attn.k_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.self_attn.q_norm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.self_attn.q_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.4.self_attn.v_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.mlp.experts.0.down_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.mlp.experts.0.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.mlp.experts.0.up_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.mlp.experts.1.down_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.mlp.experts.1.gate_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.mlp.experts.1.up_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.mlp.gate.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.post_feedforward_layernorm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.self_attn.k_norm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.self_attn.k_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.self_attn.q_norm.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.self_attn.q_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.5.self_attn.v_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.6.mlp.experts.0.down_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.mlp.experts.0.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.mlp.experts.0.up_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.mlp.experts.1.down_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.mlp.experts.1.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.mlp.experts.1.up_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.mlp.gate.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.post_attention_layernorm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.post_feedforward_layernorm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.self_attn.k_norm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.6.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.self_attn.q_norm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00010.safetensors",
+    "model.layers.7.mlp.experts.0.down_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.mlp.experts.0.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.mlp.experts.0.up_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.mlp.experts.1.down_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.mlp.experts.1.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.mlp.experts.1.up_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.mlp.gate.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.post_attention_layernorm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.post_feedforward_layernorm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.self_attn.k_norm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.self_attn.k_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.self_attn.q_norm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.self_attn.q_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.7.self_attn.v_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.mlp.experts.0.down_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.mlp.experts.0.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.mlp.experts.0.up_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.mlp.experts.1.down_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.mlp.experts.1.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.mlp.experts.1.up_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.mlp.gate.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.post_attention_layernorm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.post_feedforward_layernorm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.self_attn.k_norm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.self_attn.k_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.self_attn.q_norm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.self_attn.q_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.8.self_attn.v_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.mlp.experts.0.down_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.mlp.experts.0.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.mlp.experts.0.up_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.mlp.experts.1.down_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.9.mlp.experts.1.gate_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.mlp.experts.1.up_proj.weight": "model-00004-of-00010.safetensors",
+    "model.layers.9.mlp.gate.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.post_attention_layernorm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.9.post_feedforward_layernorm.weight": "model-00004-of-00010.safetensors",
+    "model.layers.9.self_attn.k_norm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.self_attn.k_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.self_attn.o_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.self_attn.q_norm.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.self_attn.q_proj.weight": "model-00003-of-00010.safetensors",
+    "model.layers.9.self_attn.v_proj.weight": "model-00003-of-00010.safetensors",
+    "model.norm.weight": "model-00010-of-00010.safetensors"
+  }
+}