Upload folder using huggingface_hub

README.md

@@ -0,0 +1,114 @@
+---
+language:
+- eng
+tags:
+- sft
+- Yi-34B-200K
+license:
+- mit
+datasets:
+- LDJnr/LessWrong-Amplify-Instruct
+- LDJnr/Pure-Dove
+- LDJnr/Verified-Camel
+---
+
+## **Nous-Capybara-34B V1.9**
+
+**This is trained on the Yi-34B model with 200K context length, for 3 epochs on the Capybara dataset!**
+
+**First 34B Nous model and first 200K context length Nous model!**
+
+The Capybara series is the first Nous collection of models made by fine-tuning mostly on data created by Nous in-house. 
+
+We leverage our novel data synthesis technique called Amplify-instruct (Paper coming soon), the seed distribution and synthesis method are comprised of a synergistic combination of top performing existing data synthesis techniques and distributions used for SOTA models such as Airoboros, Evol-Instruct(WizardLM), Orca, Vicuna, Know_Logic, Lamini, FLASK and others, all into one lean holistically formed methodology for the dataset and model. The seed instructions used for the start of synthesized conversations are largely based on highly regarded datasets like Airoboros, Know logic, EverythingLM, GPTeacher and even entirely new seed instructions derived from posts on the website LessWrong, as well as being supplemented with certain in-house multi-turn datasets like Dove(A successor to Puffin).
+
+While performing great in it's current state, the current dataset used for fine-tuning is entirely contained within 20K training examples, this is 10 times smaller than many similar performing current models, this is signficant when it comes to scaling implications for our next generation of models once we scale our novel syntheiss methods to significantly more examples.
+
+## Process of creation and special thank yous!
+
+This model was fine-tuned by Nous Research as part of the Capybara/Amplify-Instruct project led by Luigi D.(LDJ) (Paper coming soon), as well as significant dataset formation contributions by J-Supha and general compute and experimentation management by Jeffrey Q. during ablations.
+
+Special thank you to **A16Z** for sponsoring our training, as well as **Yield Protocol** for their support in financially sponsoring resources during the R&D of this project.
+
+## Thank you to those of you that have indirectly contributed!
+
+While most of the tokens within Capybara are newly synthsized and part of datasets like Puffin/Dove, we would like to credit the single-turn datasets we leveraged as seeds that are used to generate the multi-turn data as part of the Amplify-Instruct synthesis.
+
+The datasets shown in green below are datasets that we sampled from to curate seeds that are used during Amplify-Instruct synthesis for this project.
+
+Datasets in Blue are in-house curations that previously existed prior to Capybara.
+
+![Capybara](https://i.imgur.com/yB58OoD.jpeg)
+
+
+## Prompt Format
+
+The reccomended model usage is:
+
+
+Prefix: ``USER:``
+
+Suffix: ``ASSISTANT:``
+
+Stop token: ``</s>``
+
+
+## Mutli-Modality!
+
+ - We currently have a Multi-modal model based on Capybara V1.9! 
+https://huggingface.co/NousResearch/Obsidian-3B-V0.5
+it is currently only available as a 3B sized model but larger versions coming!
+
+
+## Notable Features:
+
+ - Uses Yi-34B model as the base which is trained for 200K context length!
+
+ - Over 60% of the dataset is comprised of multi-turn conversations.(Most models are still only trained for single-turn conversations and no back and forths!)
+
+ - Over 1,000 tokens average per conversation example! (Most models are trained on conversation data that is less than 300 tokens per example.)
+
+ - Able to effectively do complex summaries of advanced topics and studies. (trained on hundreds of advanced difficult summary tasks developed in-house)
+
+ - Ability to recall information upto late 2022 without internet.
+
+ - Includes a portion of conversational data synthesized from less wrong posts, discussing very in-depth details and philosophies about the nature of reality, reasoning, rationality, self-improvement and related concepts.
+
+## Example Outputs from Capybara V1.9 7B version! (examples from 34B coming soon):
+
+![Capybara](https://img001.prntscr.com/file/img001/T9yYxR1xQSaK_UGdy3t2Cw.png)
+
+![Capybara](https://img001.prntscr.com/file/img001/DQXqmKbsQQOIcgny1eoGNA.png)
+
+![Capybara](https://img001.prntscr.com/file/img001/85X3L9ZxTsOKo3fUQ7GRVA.png)
+
+## Benchmarks! (Coming soon!)
+ 
+
+## Future model sizes
+
+Capybara V1.9 now currently has a 3B, 7B and 34B size, and we plan to eventually have a 13B and 70B version in the future, as well as a potential 1B version based on phi-1.5 or Tiny Llama.
+
+## How you can help!
+
+In the near future we plan on leveraging the help of domain specific expert volunteers to eliminate any mathematically/verifiably incorrect answers from our training curations. 
+
+If you have at-least a bachelors in mathematics, physics, biology or chemistry and would like to volunteer even just 30 minutes of your expertise time, please contact LDJ on discord!
+
+## Dataset contamination.
+
+We have checked the capybara dataset for contamination for several of the most popular datasets and can confirm that there is no contaminaton found.
+
+We leveraged minhash to check for 100%, 99%, 98% and 97% similarity matches between our data and the questions and answers in benchmarks, we found no exact matches, nor did we find any matches down to the 97% similarity level.
+
+The following are benchmarks we checked for contamination against our dataset:
+
+- HumanEval
+
+- AGIEval
+
+- TruthfulQA
+
+- MMLU
+
+- GPT4All

config.json

@@ -0,0 +1,28 @@
+{
+  "_name_or_path": "larryvrh/Yi-34B-200K-Llamafied",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 7168,
+  "initializer_range": 0.02,
+  "intermediate_size": 20480,
+  "max_position_embeddings": 200000,
+  "model_type": "llama",
+  "num_attention_heads": 56,
+  "num_hidden_layers": 60,
+  "num_key_value_heads": 8,
+  "pad_token_id": 0,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 5000000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.34.1",
+  "use_cache": true,
+  "vocab_size": 64000
+}

generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "pad_token_id": 0,
+  "transformers_version": "4.34.1"
+}

model-00001-of-00007.safetensors

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

model-00002-of-00007.safetensors

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

model-00003-of-00007.safetensors

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

model-00004-of-00007.safetensors

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

model-00005-of-00007.safetensors

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

model-00006-of-00007.safetensors

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

model-00007-of-00007.safetensors

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

model.safetensors.index.json

@@ -0,0 +1,550 @@
+{
+    "metadata": {
+        "total_size": 68777834496
+    },
+    "weight_map": {
+        "lm_head.weight": "model-00007-of-00007.safetensors",
+        "model.embed_tokens.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.input_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.input_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.10.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.10.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.10.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.17.input_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.17.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.17.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.17.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.17.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.17.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.17.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.17.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.17.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.input_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.18.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.input_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.19.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.2.input_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.2.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.2.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.20.input_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.20.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.20.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.20.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.20.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.20.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.20.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.20.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.20.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.input_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.21.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.input_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.22.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.input_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.input_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.25.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.25.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.25.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.25.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+        "model.layers.26.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.26.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.26.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.26.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.26.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.26.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.26.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.26.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.26.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.27.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.28.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.29.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.3.input_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.3.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.3.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.30.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.30.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.30.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.30.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.30.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.30.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.30.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.30.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.30.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.31.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.32.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.input_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.33.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.34.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.34.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.34.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.34.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.34.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.34.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.34.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.34.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.34.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+        "model.layers.35.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.35.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.35.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.35.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.35.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.35.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.35.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.35.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.35.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.36.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.37.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.38.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.39.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.4.input_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.4.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.4.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.40.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.40.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.40.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.40.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.40.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.40.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.40.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.40.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.40.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.41.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.input_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.42.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.43.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.43.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.43.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.43.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.43.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.43.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.43.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.43.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.43.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+        "model.layers.44.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.44.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.44.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.44.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.44.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.44.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.44.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.44.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.44.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.45.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.46.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.47.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.48.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.49.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.5.input_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.5.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.5.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.50.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.50.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.50.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.50.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.50.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.50.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.50.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.50.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.50.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.input_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.51.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.52.input_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.52.mlp.down_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.52.mlp.gate_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.52.mlp.up_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.52.post_attention_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.52.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.52.self_attn.o_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.52.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.52.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+        "model.layers.53.input_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.53.mlp.down_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.53.mlp.gate_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.53.mlp.up_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.53.post_attention_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.53.self_attn.k_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.53.self_attn.o_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.53.self_attn.q_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.53.self_attn.v_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.input_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.mlp.down_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.mlp.gate_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.mlp.up_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.post_attention_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.self_attn.k_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.self_attn.o_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.self_attn.q_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.54.self_attn.v_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.input_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.mlp.down_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.mlp.gate_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.mlp.up_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.post_attention_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.self_attn.k_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.self_attn.o_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.self_attn.q_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.55.self_attn.v_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.input_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.mlp.down_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.mlp.gate_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.mlp.up_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.post_attention_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.self_attn.k_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.self_attn.o_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.self_attn.q_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.56.self_attn.v_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.input_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.mlp.down_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.mlp.gate_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.mlp.up_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.post_attention_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.self_attn.k_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.self_attn.o_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.self_attn.q_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.57.self_attn.v_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.input_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.mlp.down_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.mlp.gate_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.mlp.up_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.post_attention_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.self_attn.k_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.self_attn.o_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.self_attn.q_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.58.self_attn.v_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.input_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.mlp.down_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.mlp.gate_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.mlp.up_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.post_attention_layernorm.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.self_attn.k_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.self_attn.o_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.self_attn.q_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.59.self_attn.v_proj.weight": "model-00007-of-00007.safetensors",
+        "model.layers.6.input_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.6.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.6.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.input_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.8.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.8.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.8.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.8.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+        "model.layers.9.input_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.9.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.9.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.9.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+        "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+        "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+        "model.norm.weight": "model-00007-of-00007.safetensors"
+    }
+}

special_tokens_map.json

@@ -0,0 +1,30 @@
+{
+  "bos_token": {
+    "content": "<|startoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenization_yi.py

@@ -0,0 +1,255 @@
+import os
+from shutil import copyfile
+from typing import Any, Dict, List, Optional, Tuple
+
+import sentencepiece as spm
+from transformers.tokenization_utils import AddedToken, PreTrainedTokenizer
+from transformers.utils import logging
+
+logger = logging.get_logger(__name__)
+
+VOCAB_FILES_NAMES = {"vocab_file": "tokenizer.model"}
+
+PRETRAINED_VOCAB_FILES_MAP = {
+    "vocab_file": {},
+    "tokenizer_file": {},
+}
+PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES = {}
+
+
+class YiTokenizer(PreTrainedTokenizer):
+    """
+    Construct a Yi tokenizer. Based on byte-level Byte-Pair-Encoding.
+
+    Args:
+        vocab_file (`str`):
+            Path to the vocabulary file.
+    """
+
+    vocab_files_names = VOCAB_FILES_NAMES
+    pretrained_vocab_files_map = PRETRAINED_VOCAB_FILES_MAP
+    max_model_input_sizes = PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
+    model_input_names = ["input_ids", "attention_mask"]
+
+    def __init__(
+        self,
+        vocab_file,
+        unk_token="<unk>",
+        bos_token="<|startoftext|>",
+        eos_token="<|endoftext|>",
+        pad_token="<unk>",
+        sp_model_kwargs: Optional[Dict[str, Any]] = None,
+        add_bos_token=True,
+        add_eos_token=False,
+        clean_up_tokenization_spaces=False,
+        **kwargs,
+    ):
+        self.sp_model_kwargs = {} if sp_model_kwargs is None else sp_model_kwargs
+        bos_token = (
+            AddedToken(bos_token, lstrip=False, rstrip=False)
+            if isinstance(bos_token, str)
+            else bos_token
+        )
+        eos_token = (
+            AddedToken(eos_token, lstrip=False, rstrip=False)
+            if isinstance(eos_token, str)
+            else eos_token
+        )
+        unk_token = (
+            AddedToken(unk_token, lstrip=False, rstrip=False)
+            if isinstance(unk_token, str)
+            else unk_token
+        )
+        pad_token = (
+            AddedToken(pad_token, lstrip=False, rstrip=False)
+            if isinstance(pad_token, str)
+            else pad_token
+        )
+        self.vocab_file = vocab_file
+        self.add_bos_token = add_bos_token
+        self.add_eos_token = add_eos_token
+        self.sp_model = spm.SentencePieceProcessor(**self.sp_model_kwargs)
+        self.sp_model.Load(vocab_file)
+        super().__init__(
+            bos_token=bos_token,
+            eos_token=eos_token,
+            unk_token=unk_token,
+            pad_token=pad_token,
+            add_bos_token=add_bos_token,
+            add_eos_token=add_eos_token,
+            sp_model_kwargs=self.sp_model_kwargs,
+            clean_up_tokenization_spaces=clean_up_tokenization_spaces,
+            **kwargs,
+        )
+
+    def __getstate__(self):
+        state = self.__dict__.copy()
+        state["sp_model"] = None
+        return state
+
+    def __setstate__(self, d):
+        self.__dict__ = d
+        self.sp_model = spm.SentencePieceProcessor(**self.sp_model_kwargs)
+        self.sp_model.Load(self.vocab_file)
+
+    @property
+    def vocab_size(self):
+        """Returns vocab size"""
+        return self.sp_model.get_piece_size()
+
+    def get_vocab(self):
+        """Returns vocab as a dict"""
+        vocab = {self.convert_ids_to_tokens(i): i for i in range(self.vocab_size)}
+        vocab.update(self.added_tokens_encoder)
+        return vocab
+
+    def _tokenize(self, text):
+        """Returns a tokenized string."""
+        return self.sp_model.encode(text, out_type=str)
+
+    def _convert_token_to_id(self, token):
+        """Converts a token (str) in an id using the vocab."""
+        return self.sp_model.piece_to_id(token)
+
+    def _convert_id_to_token(self, index):
+        """Converts an index (integer) in a token (str) using the vocab."""
+        token = self.sp_model.IdToPiece(index)
+        return token
+
+    def convert_tokens_to_string(self, tokens):
+        """Converts a sequence of tokens (string) in a single string."""
+        current_sub_tokens = []
+        out_string = ""
+        prev_is_special = False
+        for i, token in enumerate(tokens):
+            # make sure that special tokens are not decoded using sentencepiece model
+            if token in self.all_special_tokens:
+                if not prev_is_special and i != 0:
+                    out_string += " "
+                out_string += self.sp_model.decode(current_sub_tokens) + token
+                prev_is_special = True
+                current_sub_tokens = []
+            else:
+                current_sub_tokens.append(token)
+                prev_is_special = False
+        out_string += self.sp_model.decode(current_sub_tokens)
+        return out_string
+
+    def save_vocabulary(
+        self, save_directory, filename_prefix: Optional[str] = None
+    ) -> Tuple[str]:
+        """
+        Save the vocabulary and special tokens file to a directory.
+
+        Args:
+            save_directory (`str`):
+                The directory in which to save the vocabulary.
+
+        Returns:
+            `Tuple(str)`: Paths to the files saved.
+        """
+        if not os.path.isdir(save_directory):
+            logger.error(f"Vocabulary path ({save_directory}) should be a directory")
+            return
+        out_vocab_file = os.path.join(
+            save_directory,
+            (filename_prefix + "-" if filename_prefix else "")
+            + VOCAB_FILES_NAMES["vocab_file"],
+        )
+
+        if os.path.abspath(self.vocab_file) != os.path.abspath(
+            out_vocab_file
+        ) and os.path.isfile(self.vocab_file):
+            copyfile(self.vocab_file, out_vocab_file)
+        elif not os.path.isfile(self.vocab_file):
+            with open(out_vocab_file, "wb") as fi:
+                content_spiece_model = self.sp_model.serialized_model_proto()
+                fi.write(content_spiece_model)
+
+        return (out_vocab_file,)
+
+    def build_inputs_with_special_tokens(self, token_ids_0, token_ids_1=None):
+        bos_token_id = [self.bos_token_id] if self.add_bos_token else []
+        eos_token_id = [self.eos_token_id] if self.add_eos_token else []
+
+        output = bos_token_id + token_ids_0 + eos_token_id
+
+        if token_ids_1 is not None:
+            output = output + bos_token_id + token_ids_1 + eos_token_id
+
+        return output
+
+    def get_special_tokens_mask(
+        self,
+        token_ids_0: List[int],
+        token_ids_1: Optional[List[int]] = None,
+        already_has_special_tokens: bool = False,
+    ) -> List[int]:
+        """
+        Retrieve sequence ids from a token list that has no special tokens added. This method is called when adding
+        special tokens using the tokenizer `prepare_for_model` method.
+
+        Args:
+            token_ids_0 (`List[int]`):
+                List of IDs.
+            token_ids_1 (`List[int]`, *optional*):
+                Optional second list of IDs for sequence pairs.
+            already_has_special_tokens (`bool`, *optional*, defaults to `False`):
+                Whether or not the token list is already formatted with special tokens for the model.
+
+        Returns:
+            `List[int]`: A list of integers in the range [0, 1]: 1 for a special token, 0 for a sequence token.
+        """
+        if already_has_special_tokens:
+            return super().get_special_tokens_mask(
+                token_ids_0=token_ids_0,
+                token_ids_1=token_ids_1,
+                already_has_special_tokens=True,
+            )
+
+        bos_token_id = [1] if self.add_bos_token else []
+        eos_token_id = [1] if self.add_eos_token else []
+
+        if token_ids_1 is None:
+            return bos_token_id + ([0] * len(token_ids_0)) + eos_token_id
+        return (
+            bos_token_id
+            + ([0] * len(token_ids_0))
+            + eos_token_id
+            + bos_token_id
+            + ([0] * len(token_ids_1))
+            + eos_token_id
+        )
+
+    def create_token_type_ids_from_sequences(
+        self, token_ids_0: List[int], token_ids_1: Optional[List[int]] = None
+    ) -> List[int]:
+        """
+        Creates a mask from the two sequences passed to be used in a sequence-pair classification task. An ALBERT
+        sequence pair mask has the following format:
+
+        ```
+        0 0 0 0 0 0 0 0 0 0 0 1 1 1 1 1 1 1 1 1
+        | first sequence    | second sequence |
+        ```
+
+        if token_ids_1 is None, only returns the first portion of the mask (0s).
+
+        Args:
+            token_ids_0 (`List[int]`):
+                List of ids.
+            token_ids_1 (`List[int]`, *optional*):
+                Optional second list of IDs for sequence pairs.
+
+        Returns:
+            `List[int]`: List of [token type IDs](../glossary#token-type-ids) according to the given sequence(s).
+        """
+        bos_token_id = [self.bos_token_id] if self.add_bos_token else []
+        eos_token_id = [self.eos_token_id] if self.add_eos_token else []
+
+        output = [0] * len(bos_token_id + token_ids_0 + eos_token_id)
+
+        if token_ids_1 is not None:
+            output += [1] * len(bos_token_id + token_ids_1 + eos_token_id)
+
+        return output

tokenizer.model

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

tokenizer_config.json

@@ -0,0 +1,44 @@
+{
+  "add_bos_token": false,
+  "add_eos_token": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<|startoftext|>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "auto_map": {
+    "AutoTokenizer": [
+      "tokenization_yi.YiTokenizer",
+      null
+    ]
+  },
+  "bos_token": "<|startoftext|>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|endoftext|>",
+  "model_max_length": 200000,
+  "pad_token": "<unk>",
+  "sp_model_kwargs": {},
+  "tokenizer_class": "YiTokenizer",
+  "unk_token": "<unk>"
+}