upload

.gitattributes copy

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

LICENSE.txt

@@ -0,0 +1,126 @@
+LLAMA 2 COMMUNITY LICENSE AGREEMENT	
+Llama 2 Version Release Date: July 18, 2023
+
+"Agreement" means the terms and conditions for use, reproduction, distribution and 
+modification of the Llama Materials set forth herein.
+
+"Documentation" means the specifications, manuals and documentation 
+accompanying Llama 2 distributed by Meta at ai.meta.com/resources/models-and-
+libraries/llama-downloads/.
+
+"Licensee" or "you" means you, or your employer or any other person or entity (if 
+you are entering into this Agreement on such person or entity's behalf), of the age 
+required under applicable laws, rules or regulations to provide legal consent and that 
+has legal authority to bind your employer or such other person or entity if you are 
+entering in this Agreement on their behalf.
+
+"Llama 2" means the foundational large language models and software and 
+algorithms, including machine-learning model code, trained model weights, 
+inference-enabling code, training-enabling code, fine-tuning enabling code and other 
+elements of the foregoing distributed by Meta at ai.meta.com/resources/models-and-
+libraries/llama-downloads/.
+
+"Llama Materials" means, collectively, Meta's proprietary Llama 2 and 
+Documentation (and any portion thereof) made available under this Agreement.
+
+"Meta" or "we" means Meta Platforms Ireland Limited (if you are located in or, if you 
+are an entity, your principal place of business is in the EEA or Switzerland) and Meta 
+Platforms, Inc. (if you are located outside of the EEA or Switzerland). 
+
+By clicking "I Accept" below or by using or distributing any portion or element of the 
+Llama Materials, you agree to be bound by this Agreement.
+
+1. License Rights and Redistribution. 
+
+      a. Grant of Rights. You are granted a non-exclusive, worldwide, non-
+transferable and royalty-free limited license under Meta's intellectual property or 
+other rights owned by Meta embodied in the Llama Materials to use, reproduce, 
+distribute, copy, create derivative works of, and make modifications to the Llama 
+Materials.  
+      
+      b. Redistribution and Use.  
+
+            i. If you distribute or make the Llama Materials, or any derivative works 
+thereof, available to a third party, you shall provide a copy of this Agreement to such 
+third party. 
+            ii.  If you receive Llama Materials, or any derivative works thereof, from 
+a Licensee as part of an integrated end user product, then Section 2 of this 
+Agreement will not apply to you. 
+
+            iii. You must retain in all copies of the Llama Materials that you 
+distribute the following attribution notice within a "Notice" text file distributed as a 
+part of such copies: "Llama 2 is licensed under the LLAMA 2 Community License, 
+Copyright (c) Meta Platforms, Inc. All Rights Reserved."
+
+            iv. Your use of the Llama Materials must comply with applicable laws 
+and regulations (including trade compliance laws and regulations) and adhere to the 
+Acceptable Use Policy for the Llama Materials (available at 
+https://ai.meta.com/llama/use-policy), which is hereby incorporated by reference into 
+this Agreement.
+
+            v. You will not use the Llama Materials or any output or results of the 
+Llama Materials to improve any other large language model (excluding Llama 2 or 
+derivative works thereof).  
+
+2. Additional Commercial Terms. If, on the Llama 2 version release date, the 
+monthly active users of the products or services made available by or for Licensee, 
+or Licensee's affiliates, is greater than 700 million monthly active users in the 
+preceding calendar month, you must request a license from Meta, which Meta may 
+grant to you in its sole discretion, and you are not authorized to exercise any of the 
+rights under this Agreement unless or until Meta otherwise expressly grants you 
+such rights.
+            
+3. Disclaimer of Warranty. UNLESS REQUIRED BY APPLICABLE LAW, THE 
+LLAMA MATERIALS AND ANY OUTPUT AND RESULTS THEREFROM ARE 
+PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, 
+EITHER EXPRESS OR IMPLIED, INCLUDING, WITHOUT LIMITATION, ANY 
+WARRANTIES OF TITLE, NON-INFRINGEMENT, MERCHANTABILITY, OR 
+FITNESS FOR A PARTICULAR PURPOSE. YOU ARE SOLELY RESPONSIBLE 
+FOR DETERMINING THE APPROPRIATENESS OF USING OR REDISTRIBUTING 
+THE LLAMA MATERIALS AND ASSUME ANY RISKS ASSOCIATED WITH YOUR 
+USE OF THE LLAMA MATERIALS AND ANY OUTPUT AND RESULTS.
+
+4. Limitation of Liability. IN NO EVENT WILL META OR ITS AFFILIATES BE 
+LIABLE UNDER ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, TORT, 
+NEGLIGENCE, PRODUCTS LIABILITY, OR OTHERWISE, ARISING OUT OF THIS 
+AGREEMENT, FOR ANY LOST PROFITS OR ANY INDIRECT, SPECIAL, 
+CONSEQUENTIAL, INCIDENTAL, EXEMPLARY OR PUNITIVE DAMAGES, EVEN 
+IF META OR ITS AFFILIATES HAVE BEEN ADVISED OF THE POSSIBILITY OF 
+ANY OF THE FOREGOING.
+ 
+5. Intellectual Property.
+
+      a. No trademark licenses are granted under this Agreement, and in 
+connection with the Llama Materials, neither Meta nor Licensee may use any name 
+or mark owned by or associated with the other or any of its affiliates, except as 
+required for reasonable and customary use in describing and redistributing the 
+Llama Materials.
+
+      b. Subject to Meta's ownership of Llama Materials and derivatives made by or 
+for Meta, with respect to any derivative works and modifications of the Llama 
+Materials that are made by you, as between you and Meta, you are and will be the 
+owner of such derivative works and modifications.
+
+      c. If you institute litigation or other proceedings against Meta or any entity 
+(including a cross-claim or counterclaim in a lawsuit) alleging that the Llama 
+Materials or Llama 2 outputs or results, or any portion of any of the foregoing, 
+constitutes infringement of intellectual property or other rights owned or licensable 
+by you, then any licenses granted to you under this Agreement shall terminate as of 
+the date such litigation or claim is filed or instituted. You will indemnify and hold 
+harmless Meta from and against any claim by any third party arising out of or related 
+to your use or distribution of the Llama Materials.
+
+6. Term and Termination. The term of this Agreement will commence upon your 
+acceptance of this Agreement or access to the Llama Materials and will continue in 
+full force and effect until terminated in accordance with the terms and conditions 
+herein. Meta may terminate this Agreement if you are in breach of any term or 
+condition of this Agreement. Upon termination of this Agreement, you shall delete 
+and cease use of the Llama Materials. Sections 3, 4 and 7 shall survive the 
+termination of this Agreement. 
+
+7. Governing Law and Jurisdiction. This Agreement will be governed and 
+construed under the laws of the State of California without regard to choice of law 
+principles, and the UN Convention on Contracts for the International Sale of Goods 
+does not apply to this Agreement. The courts of California shall have exclusive 
+jurisdiction of any dispute arising out of this Agreement. 
+

README.md

@@ -0,0 +1,323 @@
+---
+extra_gated_heading: You need to share contact information with Meta to access this model
+extra_gated_prompt: >-
+  ### LLAMA 2 COMMUNITY LICENSE AGREEMENT
+
+  "Agreement" means the terms and conditions for use, reproduction, distribution
+  and  modification of the Llama Materials set forth herein. 
+
+  "Documentation" means the specifications, manuals and documentation 
+  accompanying Llama 2 distributed by Meta at
+  https://ai.meta.com/resources/models-and-libraries/llama-downloads/.  
+
+  "Licensee" or "you" means you, or your employer or any other person or entity
+  (if you are entering into this Agreement on such person or entity's behalf),
+  of the age required under applicable laws, rules or regulations to provide
+  legal consent and that has legal authority to bind your employer or such other
+  person or  entity if you are  entering in this Agreement on their behalf. 
+
+  "Llama 2" means the foundational large language models and software and
+  algorithms, including machine-learning model code, trained model weights,
+  inference-enabling code, training-enabling code, fine-tuning enabling code and
+  other  elements of the foregoing distributed by Meta at
+  ai.meta.com/resources/models-and-libraries/llama-downloads/.
+
+  "Llama Materials" means, collectively, Meta's proprietary Llama 2 and
+  documentation (and any portion thereof) made available under this Agreement.
+
+  "Meta" or "we" means Meta Platforms Ireland Limited (if you are located in or,
+  if you are an entity, your principal place of business is in the EEA or
+  Switzerland) and Meta Platforms, Inc. (if you are located outside of the EEA
+  or Switzerland). 
+
+
+  By clicking "I Accept" below or by using or distributing any portion or
+  element of the Llama Materials, you agree to be bound by this Agreement.
+
+  1. License Rights and Redistribution. 
+
+  a. Grant of Rights. You are granted a non-exclusive, worldwide, non-
+  transferable and royalty-free limited license under Meta's intellectual
+  property or  other rights owned by Meta embodied in the Llama Materials to
+  use, reproduce,  distribute, copy, create derivative works of, and make
+  modifications to the Llama  Materials.  
+
+  b. Redistribution and Use.
+
+  i. If you distribute or make the Llama Materials, or any derivative works 
+  thereof, available to a third party, you shall provide a copy of this
+  Agreement to such  third party. 
+
+  ii.  If you receive Llama Materials, or any derivative works thereof, from  a
+  Licensee as part of an integrated end user product, then Section 2 of this 
+  Agreement will not apply to you. 
+
+  iii. You must retain in all copies of the Llama Materials that you  distribute
+  the following attribution notice within a "Notice" text file distributed as a 
+  part of such copies: "Llama 2 is licensed under the LLAMA 2 Community
+  License,  Copyright (c) Meta Platforms, Inc. All Rights Reserved."
+
+  iv. Your use of the Llama Materials must comply with applicable laws  and
+  regulations (including trade compliance laws and regulations) and adhere to
+  the  Acceptable Use Policy for the Llama Materials (available at 
+  https://ai.meta.com/llama/use-policy), which is hereby incorporated by
+  reference into  this Agreement.
+
+  v. You will not use the Llama Materials or any output or results of the  Llama
+  Materials to improve any other large language model (excluding Llama 2 or 
+  derivative works thereof).  
+
+
+  2. Additional Commercial Terms. If, on the Llama 2 version release date, the 
+  monthly active users of the products or services made available by or for
+  Licensee,  or Licensee's affiliates, is greater than 700 million monthly
+  active users in the  preceding calendar month, you must request a license from
+  Meta, which Meta may  grant to you in its sole discretion, and you are not
+  authorized to exercise any of the  rights under this Agreement unless or until
+  Meta otherwise expressly grants you  such rights.
+
+  3. Disclaimer of Warranty. UNLESS REQUIRED BY APPLICABLE LAW, THE  LLAMA
+  MATERIALS AND ANY OUTPUT AND RESULTS THEREFROM ARE  PROVIDED ON AN "AS IS"
+  BASIS, WITHOUT WARRANTIES OF ANY KIND,  EITHER EXPRESS OR IMPLIED, INCLUDING,
+  WITHOUT LIMITATION, ANY  WARRANTIES OF TITLE, NON-INFRINGEMENT,
+  MERCHANTABILITY, OR  FITNESS FOR A PARTICULAR PURPOSE. YOU ARE SOLELY
+  RESPONSIBLE  FOR DETERMINING THE APPROPRIATENESS OF USING OR REDISTRIBUTING 
+  THE LLAMA MATERIALS AND ASSUME ANY RISKS ASSOCIATED WITH YOUR  USE OF THE
+  LLAMA MATERIALS AND ANY OUTPUT AND RESULTS.
+
+  4. Limitation of Liability. IN NO EVENT WILL META OR ITS AFFILIATES BE  LIABLE
+  UNDER ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, TORT,  NEGLIGENCE,
+  PRODUCTS LIABILITY, OR OTHERWISE, ARISING OUT OF THIS  AGREEMENT, FOR ANY LOST
+  PROFITS OR ANY INDIRECT, SPECIAL,  CONSEQUENTIAL, INCIDENTAL, EXEMPLARY OR
+  PUNITIVE DAMAGES, EVEN  IF META OR ITS AFFILIATES HAVE BEEN ADVISED OF THE
+  POSSIBILITY OF  ANY OF THE FOREGOING.
+
+
+  5. Intellectual Property.
+
+  a. No trademark licenses are granted under this Agreement, and in  connection
+  with the Llama Materials, neither Meta nor Licensee may use any name  or mark
+  owned by or associated with the other or any of its affiliates, except as 
+  required for reasonable and customary use in describing and redistributing
+  the  Llama Materials.
+
+  b. Subject to Meta's ownership of Llama Materials and derivatives made by or 
+  for Meta, with respect to any derivative works and modifications of the Llama 
+  Materials that are made by you, as between you and Meta, you are and will be
+  the  owner of such derivative works and modifications.
+
+  c. If you institute litigation or other proceedings against Meta or any
+  entity  (including a cross-claim or counterclaim in a lawsuit) alleging that
+  the Llama  Materials or Llama 2 outputs or results, or any portion of any of
+  the foregoing,  constitutes infringement of intellectual property or other
+  rights owned or licensable  by you, then any licenses granted to you under
+  this Agreement shall terminate as of  the date such litigation or claim is
+  filed or instituted. You will indemnify and hold  harmless Meta from and
+  against any claim by any third party arising out of or related  to your use or
+  distribution of the Llama Materials.
+
+  6. Term and Termination. The term of this Agreement will commence upon your 
+  acceptance of this Agreement or access to the Llama Materials and will
+  continue in  full force and effect until terminated in accordance with the
+  terms and conditions  herein. Meta may terminate this Agreement if you are in
+  breach of any term or  condition of this Agreement. Upon termination of this
+  Agreement, you shall delete  and cease use of the Llama Materials. Sections 3,
+  4 and 7 shall survive the  termination of this Agreement. 
+
+  7. Governing Law and Jurisdiction. This Agreement will be governed and 
+  construed under the laws of the State of California without regard to choice
+  of law  principles, and the UN Convention on Contracts for the International
+  Sale of Goods  does not apply to this Agreement. The courts of California
+  shall have exclusive  jurisdiction of any dispute arising out of this
+  Agreement. 
+
+  ### Llama 2 Acceptable Use Policy
+
+  Meta is committed to promoting safe and fair use of its tools and features,
+  including Llama 2. If you access or use Llama 2, you agree to this Acceptable
+  Use Policy (“Policy”). The most recent copy of this policy can be found at
+  [ai.meta.com/llama/use-policy](http://ai.meta.com/llama/use-policy).
+
+  #### Prohibited Uses
+
+  We want everyone to use Llama 2 safely and responsibly. You agree you will not
+  use, or allow others to use, Llama 2 to:
+
+  1. Violate the law or others’ rights, including to:
+        1. Engage in, promote, generate, contribute to, encourage, plan, incite, or further illegal or unlawful activity or content, such as: 
+            1. Violence or terrorism 
+            2. Exploitation or harm to children, including the solicitation, creation, acquisition, or dissemination of child exploitative content or failure to report Child Sexual Abuse Material
+            3. Human trafficking, exploitation, and sexual violence
+            4. The illegal distribution of information or materials to minors, including obscene materials, or failure to employ legally required age-gating in connection with such information or materials.
+            5. Sexual solicitation
+            6. Any other criminal activity
+        2. Engage in, promote, incite, or facilitate the harassment, abuse, threatening, or bullying of individuals or groups of individuals
+        3. Engage in, promote, incite, or facilitate discrimination or other unlawful or harmful conduct in the provision of employment, employment benefits, credit, housing, other economic benefits, or other essential goods and services
+        4. Engage in the unauthorized or unlicensed practice of any profession including, but not limited to, financial, legal, medical/health, or related professional practices 
+        5. Collect, process, disclose, generate, or infer health, demographic, or other sensitive personal or private information about individuals without rights and consents required by applicable laws
+        6. Engage in or facilitate any action or generate any content that infringes, misappropriates, or otherwise violates any third-party rights, including the outputs or results of any products or services using the Llama 2 Materials
+        7. Create, generate, or facilitate the creation of malicious code, malware, computer viruses or do anything else that could disable, overburden, interfere with or impair the proper working, integrity, operation or appearance of a website or computer system 
+  2. Engage in, promote, incite, facilitate, or assist in the planning or
+  development of activities that present a risk of death or bodily harm to
+  individuals, including use of Llama 2 related to the following:
+      1. Military, warfare, nuclear industries or applications, espionage, use for materials or activities that are subject to the International Traffic Arms Regulations (ITAR) maintained by the United States Department of State
+      2. Guns and illegal weapons (including weapon development)
+      3. Illegal drugs and regulated/controlled substances
+      4. Operation of critical infrastructure, transportation technologies, or heavy machinery
+      5. Self-harm or harm to others, including suicide, cutting, and eating disorders
+      6. Any content intended to incite or promote violence, abuse, or any infliction of bodily harm to an individual
+  3. Intentionally deceive or mislead others, including use of Llama 2 related
+  to the following:
+      1. Generating, promoting, or furthering fraud or the creation or promotion of disinformation
+      2. Generating, promoting, or furthering defamatory content, including the creation of defamatory statements, images, or other content
+      3. Generating, promoting, or further distributing spam
+      4. Impersonating another individual without consent, authorization, or legal right
+      5. Representing that the use of Llama 2 or outputs are human-generated
+      6. Generating or facilitating false online engagement, including fake reviews and other means of fake online engagement 
+      4. Fail to appropriately disclose to end users any known dangers of your AI system 
+  Please report any violation of this Policy, software “bug,” or other problems
+  that could lead to a violation of this Policy through one of the following
+  means: 
+      * Reporting issues with the model: [github.com/facebookresearch/llama](http://github.com/facebookresearch/llama)
+      * Reporting risky content generated by the model: [developers.facebook.com/llama_output_feedback](http://developers.facebook.com/llama_output_feedback)
+      * Reporting bugs and security concerns: [facebook.com/whitehat/info](http://facebook.com/whitehat/info) 
+      * Reporting violations of the Acceptable Use Policy or unlicensed uses of Llama: [[email protected]](mailto:[email protected])
+extra_gated_fields:
+  First Name: text
+  Last Name: text
+  Date of birth: date_picker
+  Country: country
+  Affiliation: text
+  geo: ip_location
+  By clicking Submit below I accept the terms of the license and acknowledge that the information I provide will be collected stored processed and shared in accordance with the Meta Privacy Policy: checkbox
+extra_gated_description: >-
+  The information you provide will be collected, stored, processed and shared in
+  accordance with the [Meta Privacy
+  Policy](https://www.facebook.com/privacy/policy/).
+extra_gated_button_content: Submit
+language:
+- en
+pipeline_tag: text-generation
+tags:
+- facebook
+- meta
+- pytorch
+- llama
+- llama-2
+license: llama2
+---
+# **Llama 2**
+Llama 2 is a collection of pretrained and fine-tuned generative text models ranging in scale from 7 billion to 70 billion parameters. This is the repository for the 70B pretrained model, converted for the Hugging Face Transformers format. Links to other models can be found in the index at the bottom.
+
+## Model Details
+*Note: Use of this model is governed by the Meta license. In order to download the model weights and tokenizer, please visit the [website](https://ai.meta.com/resources/models-and-libraries/llama-downloads/) and accept our License before requesting access here.*
+
+Meta developed and publicly released the Llama 2 family of large language models (LLMs), a collection of pretrained and fine-tuned generative text models ranging in scale from 7 billion to 70 billion parameters. Our fine-tuned LLMs, called Llama-2-Chat, are optimized for dialogue use cases. Llama-2-Chat models outperform open-source chat models on most benchmarks we tested, and in our human evaluations for helpfulness and safety, are on par with some popular closed-source models like ChatGPT and PaLM.
+
+**Model Developers** Meta
+
+**Variations** Llama 2 comes in a range of parameter sizes — 7B, 13B, and 70B — as well as pretrained and fine-tuned variations.
+
+**Input** Models input text only.
+
+**Output** Models generate text only.
+
+**Model Architecture** Llama 2 is an auto-regressive language model that uses an optimized transformer architecture. The tuned versions use supervised fine-tuning (SFT) and reinforcement learning with human feedback (RLHF) to align to human preferences for helpfulness and safety.
+
+
+||Training Data|Params|Content Length|GQA|Tokens|LR|
+|---|---|---|---|---|---|---|
+|Llama 2|*A new mix of publicly available online data*|7B|4k|&#10007;|2.0T|3.0 x 10<sup>-4</sup>|
+|Llama 2|*A new mix of publicly available online data*|13B|4k|&#10007;|2.0T|3.0 x 10<sup>-4</sup>|
+|Llama 2|*A new mix of publicly available online data*|70B|4k|&#10004;|2.0T|1.5 x 10<sup>-4</sup>|
+
+*Llama 2 family of models.* Token counts refer to pretraining data only. All models are trained with a global batch-size of 4M tokens. Bigger models -  70B -- use Grouped-Query Attention (GQA) for improved inference scalability.
+
+**Model Dates** Llama 2 was trained between January 2023 and July 2023.
+
+**Status** This is a static model trained on an offline dataset. Future versions of the tuned models will be released as we improve model safety with community feedback.
+
+**License** A custom commercial license is available at: [https://ai.meta.com/resources/models-and-libraries/llama-downloads/](https://ai.meta.com/resources/models-and-libraries/llama-downloads/)
+
+**Research Paper** ["Llama-2: Open Foundation and Fine-tuned Chat Models"](arxiv.org/abs/2307.09288)
+
+## Intended Use
+**Intended Use Cases** Llama 2 is intended for commercial and research use in English. Tuned models are intended for assistant-like chat, whereas pretrained models can be adapted for a variety of natural language generation tasks.
+
+To get the expected features and performance for the chat versions, a specific formatting needs to be followed, including the `INST` and `<<SYS>>` tags, `BOS` and `EOS` tokens, and the whitespaces and breaklines in between (we recommend calling `strip()` on inputs to avoid double-spaces). See our reference code in github for details: [`chat_completion`](https://github.com/facebookresearch/llama/blob/main/llama/generation.py#L212).
+
+**Out-of-scope Uses** Use in any manner that violates applicable laws or regulations (including trade compliance laws).Use in languages other than English. Use in any other way that is prohibited by the Acceptable Use Policy and Licensing Agreement for Llama 2.
+
+## Hardware and Software
+**Training Factors** We used custom training libraries, Meta's Research Super Cluster, and production clusters for pretraining. Fine-tuning, annotation, and evaluation were also performed on third-party cloud compute.
+
+**Carbon Footprint** Pretraining utilized a cumulative 3.3M GPU hours of computation on hardware of type A100-80GB (TDP of 350-400W). Estimated total emissions were 539 tCO2eq, 100% of which were offset by Meta’s sustainability program.
+
+||Time (GPU hours)|Power Consumption (W)|Carbon Emitted(tCO<sub>2</sub>eq)|
+|---|---|---|---|
+|Llama 2 7B|184320|400|31.22|
+|Llama 2 13B|368640|400|62.44|
+|Llama 2 70B|1720320|400|291.42|
+|Total|3311616||539.00|
+
+**CO<sub>2</sub> emissions during pretraining.** Time: total GPU time required for training each model. Power Consumption: peak power capacity per GPU device for the GPUs used adjusted for power usage efficiency. 100% of the emissions are directly offset by Meta's sustainability program, and because we are openly releasing these models, the pretraining costs do not need to be incurred by others.
+
+## Training Data
+**Overview** Llama 2 was pretrained on 2 trillion tokens of data from publicly available sources. The fine-tuning data includes publicly available instruction datasets, as well as over one million new human-annotated examples. Neither the pretraining nor the fine-tuning datasets include Meta user data.
+
+**Data Freshness** The pretraining data has a cutoff of September 2022, but some tuning data is more recent, up to July 2023.
+
+## Evaluation Results
+
+In this section, we report the results for the Llama 1 and Llama 2 models on standard academic benchmarks.For all the evaluations, we use our internal evaluations library.
+
+|Model|Size|Code|Commonsense Reasoning|World Knowledge|Reading Comprehension|Math|MMLU|BBH|AGI Eval|
+|---|---|---|---|---|---|---|---|---|---|
+|Llama 1|7B|14.1|60.8|46.2|58.5|6.95|35.1|30.3|23.9|
+|Llama 1|13B|18.9|66.1|52.6|62.3|10.9|46.9|37.0|33.9|
+|Llama 1|33B|26.0|70.0|58.4|67.6|21.4|57.8|39.8|41.7|
+|Llama 1|65B|30.7|70.7|60.5|68.6|30.8|63.4|43.5|47.6|
+|Llama 2|7B|16.8|63.9|48.9|61.3|14.6|45.3|32.6|29.3|
+|Llama 2|13B|24.5|66.9|55.4|65.8|28.7|54.8|39.4|39.1|
+|Llama 2|70B|**37.5**|**71.9**|**63.6**|**69.4**|**35.2**|**68.9**|**51.2**|**54.2**|
+
+**Overall performance on grouped academic benchmarks.** *Code:* We report the average pass@1 scores of our models on HumanEval and MBPP. *Commonsense Reasoning:* We report the average of PIQA, SIQA, HellaSwag, WinoGrande, ARC easy and challenge, OpenBookQA, and CommonsenseQA. We report 7-shot results for CommonSenseQA and 0-shot results for all other benchmarks. *World Knowledge:* We evaluate the 5-shot performance on NaturalQuestions and TriviaQA and report the average. *Reading Comprehension:* For reading comprehension, we report the 0-shot average on SQuAD, QuAC, and BoolQ. *MATH:* We report the average of the GSM8K (8 shot) and MATH (4 shot) benchmarks at top 1.
+
+|||TruthfulQA|Toxigen|
+|---|---|---|---|
+|Llama 1|7B|27.42|23.00|
+|Llama 1|13B|41.74|23.08|
+|Llama 1|33B|44.19|22.57|
+|Llama 1|65B|48.71|21.77|
+|Llama 2|7B|33.29|**21.25**|
+|Llama 2|13B|41.86|26.10|
+|Llama 2|70B|**50.18**|24.60|
+
+**Evaluation of pretrained LLMs on automatic safety benchmarks.** For TruthfulQA, we present the percentage of generations that are both truthful and informative (the higher the better). For ToxiGen, we present the percentage of toxic generations (the smaller the better).
+
+
+|||TruthfulQA|Toxigen|
+|---|---|---|---|
+|Llama-2-Chat|7B|57.04|**0.00**|
+|Llama-2-Chat|13B|62.18|**0.00**|
+|Llama-2-Chat|70B|**64.14**|0.01|
+
+**Evaluation of fine-tuned LLMs on different safety datasets.** Same metric definitions as above.
+
+## Ethical Considerations and Limitations
+Llama 2 is a new technology that carries risks with use. Testing conducted to date has been in English, and has not covered, nor could it cover all scenarios. For these reasons, as with all LLMs, Llama 2’s potential outputs cannot be predicted in advance, and the model may in some instances produce inaccurate, biased or other objectionable responses to user prompts. Therefore, before deploying any applications of Llama 2, developers should perform safety testing and tuning tailored to their specific applications of the model.
+
+Please see the Responsible Use Guide available at [https://ai.meta.com/llama/responsible-use-guide/](https://ai.meta.com/llama/responsible-use-guide)
+
+## Reporting Issues
+Please report any software “bug,” or other problems with the models through one of the following means:
+- Reporting issues with the model: [github.com/facebookresearch/llama](http://github.com/facebookresearch/llama)
+- Reporting problematic content generated by the model: [developers.facebook.com/llama_output_feedback](http://developers.facebook.com/llama_output_feedback)
+- Reporting bugs and security concerns: [facebook.com/whitehat/info](http://facebook.com/whitehat/info)
+
+## Llama Model Index
+|Model|Llama2|Llama2-hf|Llama2-chat|Llama2-chat-hf|
+|---|---|---|---|---|
+|7B| [Link](https://huggingface.co/meta-llama/Llama-2-7b) | [Link](https://huggingface.co/meta-llama/Llama-2-7b-hf) | [Link](https://huggingface.co/meta-llama/Llama-2-7b-chat) | [Link](https://huggingface.co/meta-llama/Llama-2-7b-chat-hf)|
+|13B| [Link](https://huggingface.co/meta-llama/Llama-2-13b) | [Link](https://huggingface.co/meta-llama/Llama-2-13b-hf) | [Link](https://huggingface.co/meta-llama/Llama-2-13b-chat) | [Link](https://huggingface.co/meta-llama/Llama-2-13b-chat-hf)|
+|70B| [Link](https://huggingface.co/meta-llama/Llama-2-70b) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-hf) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-chat) | [Link](https://huggingface.co/meta-llama/Llama-2-70b-chat-hf)|

USE_POLICY.md

@@ -0,0 +1,50 @@
+# Llama 2 Acceptable Use Policy
+
+Meta is committed to promoting safe and fair use of its tools and features, including Llama 2. If you access or use Llama 2, you agree to this Acceptable Use Policy (“Policy”). The most recent copy of this policy can be found at [ai.meta.com/llama/use-policy](http://ai.meta.com/llama/use-policy).
+
+## Prohibited Uses
+We want everyone to use Llama 2 safely and responsibly. You agree you will not use, or allow others to use, Llama 2 to: 
+
+1. Violate the law or others’ rights, including to:
+    1. Engage in, promote, generate, contribute to, encourage, plan, incite, or further illegal or unlawful activity or content, such as: 
+        1. Violence or terrorism 
+        2. Exploitation or harm to children, including the solicitation, creation, acquisition, or dissemination of child exploitative content or failure to report Child Sexual Abuse Material
+        3. Human trafficking, exploitation, and sexual violence
+        4. The illegal distribution of information or materials to minors, including obscene materials, or failure to employ legally required age-gating in connection with such information or materials.
+        5. Sexual solicitation
+        6. Any other criminal activity
+    2. Engage in, promote, incite, or facilitate the harassment, abuse, threatening, or bullying of individuals or groups of individuals
+    3. Engage in, promote, incite, or facilitate discrimination or other unlawful or harmful conduct in the provision of employment, employment benefits, credit, housing, other economic benefits, or other essential goods and services
+    4. Engage in the unauthorized or unlicensed practice of any profession including, but not limited to, financial, legal, medical/health, or related professional practices 
+    5. Collect, process, disclose, generate, or infer health, demographic, or other sensitive personal or private information about individuals without rights and consents required by applicable laws
+    6. Engage in or facilitate any action or generate any content that infringes, misappropriates, or otherwise violates any third-party rights, including the outputs or results of any products or services using the Llama 2 Materials
+    7. Create, generate, or facilitate the creation of malicious code, malware, computer viruses or do anything else that could disable, overburden, interfere with or impair the proper working, integrity, operation or appearance of a website or computer system 
+
+
+
+2. Engage in, promote, incite, facilitate, or assist in the planning or development of activities that present a risk of death or bodily harm to individuals, including use of Llama 2 related to the following:
+    1. Military, warfare, nuclear industries or applications, espionage, use for materials or activities that are subject to the International Traffic Arms Regulations (ITAR) maintained by the United States Department of State
+    2. Guns and illegal weapons (including weapon development)
+    3. Illegal drugs and regulated/controlled substances
+    4. Operation of critical infrastructure, transportation technologies, or heavy machinery
+    5. Self-harm or harm to others, including suicide, cutting, and eating disorders
+    6. Any content intended to incite or promote violence, abuse, or any infliction of bodily harm to an individual
+
+
+
+3. Intentionally deceive or mislead others, including use of Llama 2 related to the following:
+    1. Generating, promoting, or furthering fraud or the creation or promotion of disinformation
+    2. Generating, promoting, or furthering defamatory content, including the creation of defamatory statements, images, or other content
+    3. Generating, promoting, or further distributing spam
+    4. Impersonating another individual without consent, authorization, or legal right
+    5. Representing that the use of Llama 2 or outputs are human-generated
+    6. Generating or facilitating false online engagement, including fake reviews and other means of fake online engagement 
+4. Fail to appropriately disclose to end users any known dangers of your AI system 
+
+Please report any violation of this Policy, software “bug,” or other problems that could lead to a violation of this Policy through one of the following means:
+
+* Reporting issues with the model: [github.com/facebookresearch/llama](http://github.com/facebookresearch/llama)
+* Reporting risky content generated by the model: [developers.facebook.com/llama_output_feedback](http://developers.facebook.com/llama_output_feedback)
+* Reporting bugs and security concerns: [facebook.com/whitehat/info](http://facebook.com/whitehat/info)
+* Reporting violations of the Acceptable Use Policy or unlicensed uses of Llama: [[email protected]](mailto:[email protected])
+

config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop16Attn/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop16Block/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop16MLP/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop32Attn/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop32Block/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop32MLP/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop40Attn/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop40Block/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop40MLP/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop48Attn/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop48Block/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop48MLP/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop4Attn/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop4Block/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop4MLP/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop8Attn/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop8Block/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configs/Llama2-70b-Drop8MLP/config.json

@@ -0,0 +1,76 @@
+{
+  "_name_or_path": "meta-llama/Llama-2-70b-hf",
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_dropped_llama.LlamaConfig",
+    "AutoModelForCausalLM": "modeling_dropped_llama.LlamaForCausalLM"
+  },
+  "bos_token_id": 1,
+  "drop_attn_list": [
+    22,
+    30,
+    32,
+    36,
+    43,
+    45,
+    46,
+    47,
+    48,
+    49,
+    50,
+    51,
+    52,
+    53,
+    54,
+    55,
+    56,
+    57,
+    58,
+    59,
+    60,
+    61,
+    62,
+    63,
+    64,
+    65,
+    66,
+    67,
+    68,
+    69,
+    70,
+    71,
+    72,
+    73,
+    74,
+    75,
+    76,
+    77,
+    78,
+    79
+  ],
+  "drop_mlp_list": null,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 8192,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 64,
+  "num_hidden_layers": 80,
+  "num_key_value_heads": 8,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

configuration_dropped_llama.py

@@ -0,0 +1,232 @@
+# coding=utf-8
+# Copyright 2022 EleutherAI and the HuggingFace Inc. team. All rights reserved.
+#
+# This code is based on EleutherAI's GPT-NeoX library and the GPT-NeoX
+# and OPT implementations in this library. It has been modified from its
+# original forms to accommodate minor architectural differences compared
+# to GPT-NeoX and OPT used by the Meta AI team that trained the model.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+""" transformers==4.38.1"""
+""" LLaMA model configuration"""
+from transformers.configuration_utils import PretrainedConfig
+from transformers.utils import logging
+
+
+logger = logging.get_logger(__name__)
+
+LLAMA_PRETRAINED_CONFIG_ARCHIVE_MAP = {}
+
+
+class LlamaConfig(PretrainedConfig):
+    r"""
+    This is the configuration class to store the configuration of a [`LlamaModel`]. It is used to instantiate an LLaMA
+    model according to the specified arguments, defining the model architecture. Instantiating a configuration with the
+    defaults will yield a similar configuration to that of the LLaMA-7B.
+
+    Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
+    documentation from [`PretrainedConfig`] for more information.
+
+
+    Args:
+        vocab_size (`int`, *optional*, defaults to 32000):
+            Vocabulary size of the LLaMA model. Defines the number of different tokens that can be represented by the
+            `inputs_ids` passed when calling [`LlamaModel`]
+        hidden_size (`int`, *optional*, defaults to 4096):
+            Dimension of the hidden representations.
+        intermediate_size (`int`, *optional*, defaults to 11008):
+            Dimension of the MLP representations.
+        num_hidden_layers (`int`, *optional*, defaults to 32):
+            Number of hidden layers in the Transformer decoder.
+        num_attention_heads (`int`, *optional*, defaults to 32):
+            Number of attention heads for each attention layer in the Transformer decoder.
+        num_key_value_heads (`int`, *optional*):
+            This is the number of key_value heads that should be used to implement Grouped Query Attention. If
+            `num_key_value_heads=num_attention_heads`, the model will use Multi Head Attention (MHA), if
+            `num_key_value_heads=1 the model will use Multi Query Attention (MQA) otherwise GQA is used. When
+            converting a multi-head checkpoint to a GQA checkpoint, each group key and value head should be constructed
+            by meanpooling all the original heads within that group. For more details checkout [this
+            paper](https://arxiv.org/pdf/2305.13245.pdf). If it is not specified, will default to
+            `num_attention_heads`.
+        hidden_act (`str` or `function`, *optional*, defaults to `"silu"`):
+            The non-linear activation function (function or string) in the decoder.
+        max_position_embeddings (`int`, *optional*, defaults to 2048):
+            The maximum sequence length that this model might ever be used with. Llama 1 supports up to 2048 tokens,
+            Llama 2 up to 4096, CodeLlama up to 16384.
+        initializer_range (`float`, *optional*, defaults to 0.02):
+            The standard deviation of the truncated_normal_initializer for initializing all weight matrices.
+        rms_norm_eps (`float`, *optional*, defaults to 1e-06):
+            The epsilon used by the rms normalization layers.
+        use_cache (`bool`, *optional*, defaults to `True`):
+            Whether or not the model should return the last key/values attentions (not used by all models). Only
+            relevant if `config.is_decoder=True`.
+        pad_token_id (`int`, *optional*):
+            Padding token id.
+        bos_token_id (`int`, *optional*, defaults to 1):
+            Beginning of stream token id.
+        eos_token_id (`int`, *optional*, defaults to 2):
+            End of stream token id.
+        pretraining_tp (`int`, *optional*, defaults to 1):
+            Experimental feature. Tensor parallelism rank used during pretraining. Please refer to [this
+            document](https://huggingface.co/docs/transformers/main/perf_train_gpu_many#tensor-parallelism) to understand more about it. This value is
+            necessary to ensure exact reproducibility of the pretraining results. Please refer to [this
+            issue](https://github.com/pytorch/pytorch/issues/76232).
+        tie_word_embeddings (`bool`, *optional*, defaults to `False`):
+            Whether to tie weight embeddings
+        rope_theta (`float`, *optional*, defaults to 10000.0):
+            The base period of the RoPE embeddings.
+        rope_scaling (`Dict`, *optional*):
+            Dictionary containing the scaling configuration for the RoPE embeddings. Currently supports two scaling
+            strategies: linear and dynamic. Their scaling factor must be a float greater than 1. The expected format is
+            `{"type": strategy name, "factor": scaling factor}`. When using this flag, don't update
+            `max_position_embeddings` to the expected new maximum. See the following thread for more information on how
+            these scaling strategies behave:
+            https://www.reddit.com/r/LocalLLaMA/comments/14mrgpr/dynamically_scaled_rope_further_increases/. This is an
+            experimental feature, subject to breaking API changes in future versions.
+        attention_bias (`bool`, defaults to `False`, *optional*, defaults to `False`):
+            Whether to use a bias in the query, key, value and output projection layers during self-attention.
+        attention_dropout (`float`, *optional*, defaults to 0.0):
+            The dropout ratio for the attention probabilities.
+
+    ```python
+    >>> from transformers import LlamaModel, LlamaConfig
+
+    >>> # Initializing a LLaMA llama-7b style configuration
+    >>> configuration = LlamaConfig()
+
+    >>> # Initializing a model from the llama-7b style configuration
+    >>> model = LlamaModel(configuration)
+
+    >>> # Accessing the model configuration
+    >>> configuration = model.config
+    ```"""
+
+    model_type = "llama"
+    keys_to_ignore_at_inference = ["past_key_values"]
+
+    def __init__(
+        self,
+        vocab_size=32000,
+        hidden_size=4096,
+        intermediate_size=11008,
+        num_hidden_layers=32,
+        num_attention_heads=32,
+        num_key_value_heads=None,
+        hidden_act="silu",
+        max_position_embeddings=2048,
+        initializer_range=0.02,
+        rms_norm_eps=1e-6,
+        use_cache=True,
+        pad_token_id=None,
+        bos_token_id=1,
+        eos_token_id=2,
+        pretraining_tp=1,
+        tie_word_embeddings=False,
+        rope_theta=10000.0,
+        rope_scaling=None,
+        attention_bias=False,
+        attention_dropout=0.0,
+        drop_mlp_list=None,
+        drop_attn_list=None,
+        **kwargs,
+    ):
+        self.vocab_size = vocab_size
+        self.max_position_embeddings = max_position_embeddings
+        self.hidden_size = hidden_size
+        self.intermediate_size = intermediate_size
+        self.num_hidden_layers = num_hidden_layers
+        self.num_attention_heads = num_attention_heads
+
+        #####################################################################################################################
+
+        # ✨ trans bool into int
+        new_drop_attn_list = []
+        if drop_attn_list is not None:
+            for idx in range(len(drop_attn_list)):
+                if isinstance(drop_attn_list[idx], bool):
+                    if drop_attn_list[idx] == True:
+                        new_drop_attn_list.append(idx)
+                elif isinstance(drop_attn_list[idx], int):
+                    new_drop_attn_list.append(drop_attn_list[idx])
+
+        new_drop_mlp_list = []
+        if drop_mlp_list is not None:
+            for idx in range(len(drop_mlp_list)):
+                if isinstance(drop_mlp_list[idx], bool):
+                    if drop_mlp_list[idx] == True:
+                        new_drop_mlp_list.append(idx)
+                elif isinstance(drop_mlp_list[idx], int):
+                    new_drop_mlp_list.append(drop_mlp_list[idx])
+
+        #####################################################################################################################
+
+        if new_drop_mlp_list:
+            self.drop_mlp_list = []
+            for idx in range(self.num_hidden_layers):
+                self.drop_mlp_list.append(True if idx in new_drop_mlp_list else False)
+        else:
+            self.drop_mlp_list = [False] * self.num_hidden_layers
+
+        if new_drop_attn_list:
+            self.drop_attn_list = []
+            for idx in range(self.num_hidden_layers):
+                self.drop_attn_list.append(True if idx in new_drop_attn_list else False)
+        else:
+            self.drop_attn_list = [False] * self.num_hidden_layers
+
+        #####################################################################################################################
+
+        # for backward compatibility
+        if num_key_value_heads is None:
+            num_key_value_heads = num_attention_heads
+
+        self.num_key_value_heads = num_key_value_heads
+        self.hidden_act = hidden_act
+        self.initializer_range = initializer_range
+        self.rms_norm_eps = rms_norm_eps
+        self.pretraining_tp = pretraining_tp
+        self.use_cache = use_cache
+        self.rope_theta = rope_theta
+        self.rope_scaling = rope_scaling
+        self._rope_scaling_validation()
+        self.attention_bias = attention_bias
+        self.attention_dropout = attention_dropout
+
+        super().__init__(
+            pad_token_id=pad_token_id,
+            bos_token_id=bos_token_id,
+            eos_token_id=eos_token_id,
+            tie_word_embeddings=tie_word_embeddings,
+            **kwargs,
+        )
+
+    def _rope_scaling_validation(self):
+        """
+        Validate the `rope_scaling` configuration.
+        """
+        if self.rope_scaling is None:
+            return
+
+        if not isinstance(self.rope_scaling, dict) or len(self.rope_scaling) != 2:
+            raise ValueError(
+                "`rope_scaling` must be a dictionary with with two fields, `type` and `factor`, "
+                f"got {self.rope_scaling}"
+            )
+        rope_scaling_type = self.rope_scaling.get("type", None)
+        rope_scaling_factor = self.rope_scaling.get("factor", None)
+        if rope_scaling_type is None or rope_scaling_type not in ["linear", "dynamic"]:
+            raise ValueError(
+                f"`rope_scaling`'s type field must be one of ['linear', 'dynamic'], got {rope_scaling_type}"
+            )
+        if rope_scaling_factor is None or not isinstance(rope_scaling_factor, float) or rope_scaling_factor <= 1.0:
+            raise ValueError(f"`rope_scaling`'s factor field must be a float > 1, got {rope_scaling_factor}")

generation_config.json

@@ -0,0 +1,10 @@
+{
+  "bos_token_id": 1,
+  "do_sample": true,
+  "eos_token_id": 2,
+  "pad_token_id": 0,
+  "max_length": 4096,
+  "temperature": 0.6,
+  "top_p": 0.9,
+  "transformers_version": "4.32.0.dev0"
+}

model.safetensors.index.json

@@ -0,0 +1,810 @@
+{
+  "metadata": {
+    "total_size": 137953316864
+  },
+  "weight_map": {
+    "lm_head.weight": "model-00015-of-00015.safetensors",
+    "model.embed_tokens.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.input_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.mlp.down_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.mlp.up_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.0.self_attn.rotary_emb.inv_freq": "model-00001-of-00015.safetensors",
+    "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.input_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.mlp.down_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.mlp.up_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.1.self_attn.rotary_emb.inv_freq": "model-00001-of-00015.safetensors",
+    "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.10.input_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.10.mlp.down_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.10.mlp.up_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.10.self_attn.rotary_emb.inv_freq": "model-00002-of-00015.safetensors",
+    "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.11.input_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.11.mlp.down_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.11.mlp.gate_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.11.mlp.up_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.11.post_attention_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.11.self_attn.rotary_emb.inv_freq": "model-00002-of-00015.safetensors",
+    "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.12.input_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.12.mlp.down_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.12.mlp.gate_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.12.mlp.up_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.12.post_attention_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.12.self_attn.k_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.12.self_attn.o_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.12.self_attn.q_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.12.self_attn.rotary_emb.inv_freq": "model-00003-of-00015.safetensors",
+    "model.layers.12.self_attn.v_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.input_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.mlp.down_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.mlp.gate_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.mlp.up_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.post_attention_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.self_attn.k_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.self_attn.o_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.self_attn.q_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.13.self_attn.rotary_emb.inv_freq": "model-00003-of-00015.safetensors",
+    "model.layers.13.self_attn.v_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.input_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.mlp.down_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.mlp.gate_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.mlp.up_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.post_attention_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.self_attn.k_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.self_attn.o_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.self_attn.q_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.14.self_attn.rotary_emb.inv_freq": "model-00003-of-00015.safetensors",
+    "model.layers.14.self_attn.v_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.input_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.mlp.down_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.mlp.gate_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.mlp.up_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.post_attention_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.self_attn.k_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.self_attn.o_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.self_attn.q_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.15.self_attn.rotary_emb.inv_freq": "model-00003-of-00015.safetensors",
+    "model.layers.15.self_attn.v_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.input_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.mlp.down_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.mlp.gate_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.mlp.up_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.post_attention_layernorm.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.self_attn.k_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.self_attn.o_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.self_attn.q_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.16.self_attn.rotary_emb.inv_freq": "model-00003-of-00015.safetensors",
+    "model.layers.16.self_attn.v_proj.weight": "model-00003-of-00015.safetensors",
+    "model.layers.17.input_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.17.mlp.down_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.17.mlp.gate_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.17.mlp.up_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.17.post_attention_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.17.self_attn.k_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.17.self_attn.o_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.17.self_attn.q_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.17.self_attn.rotary_emb.inv_freq": "model-00004-of-00015.safetensors",
+    "model.layers.17.self_attn.v_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.input_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.mlp.down_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.mlp.gate_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.mlp.up_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.post_attention_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.self_attn.k_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.self_attn.o_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.self_attn.q_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.18.self_attn.rotary_emb.inv_freq": "model-00004-of-00015.safetensors",
+    "model.layers.18.self_attn.v_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.input_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.mlp.down_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.mlp.gate_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.mlp.up_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.post_attention_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.self_attn.k_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.self_attn.o_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.self_attn.q_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.19.self_attn.rotary_emb.inv_freq": "model-00004-of-00015.safetensors",
+    "model.layers.19.self_attn.v_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.2.input_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.2.mlp.down_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.2.mlp.up_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.2.self_attn.rotary_emb.inv_freq": "model-00001-of-00015.safetensors",
+    "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.20.input_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.20.mlp.down_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.20.mlp.gate_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.20.mlp.up_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.20.post_attention_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.20.self_attn.k_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.20.self_attn.o_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.20.self_attn.q_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.20.self_attn.rotary_emb.inv_freq": "model-00004-of-00015.safetensors",
+    "model.layers.20.self_attn.v_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.input_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.mlp.down_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.mlp.gate_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.mlp.up_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.post_attention_layernorm.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.self_attn.k_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.self_attn.o_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.self_attn.q_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.21.self_attn.rotary_emb.inv_freq": "model-00004-of-00015.safetensors",
+    "model.layers.21.self_attn.v_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.22.input_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.22.mlp.down_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.22.mlp.gate_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.22.mlp.up_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.22.post_attention_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.22.self_attn.k_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.22.self_attn.o_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.22.self_attn.q_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.22.self_attn.rotary_emb.inv_freq": "model-00004-of-00015.safetensors",
+    "model.layers.22.self_attn.v_proj.weight": "model-00004-of-00015.safetensors",
+    "model.layers.23.input_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.23.mlp.down_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.23.mlp.gate_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.23.mlp.up_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.23.post_attention_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.23.self_attn.k_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.23.self_attn.o_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.23.self_attn.q_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.23.self_attn.rotary_emb.inv_freq": "model-00005-of-00015.safetensors",
+    "model.layers.23.self_attn.v_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.input_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.mlp.down_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.mlp.gate_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.mlp.up_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.post_attention_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.self_attn.k_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.self_attn.o_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.self_attn.q_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.24.self_attn.rotary_emb.inv_freq": "model-00005-of-00015.safetensors",
+    "model.layers.24.self_attn.v_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.input_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.mlp.down_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.mlp.gate_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.mlp.up_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.post_attention_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.self_attn.k_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.self_attn.o_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.self_attn.q_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.25.self_attn.rotary_emb.inv_freq": "model-00005-of-00015.safetensors",
+    "model.layers.25.self_attn.v_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.input_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.mlp.down_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.mlp.gate_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.mlp.up_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.post_attention_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.self_attn.k_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.self_attn.o_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.self_attn.q_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.26.self_attn.rotary_emb.inv_freq": "model-00005-of-00015.safetensors",
+    "model.layers.26.self_attn.v_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.input_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.mlp.down_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.mlp.gate_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.mlp.up_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.post_attention_layernorm.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.self_attn.k_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.self_attn.o_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.self_attn.q_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.27.self_attn.rotary_emb.inv_freq": "model-00005-of-00015.safetensors",
+    "model.layers.27.self_attn.v_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.28.input_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.28.mlp.down_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.28.mlp.gate_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.28.mlp.up_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.28.post_attention_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.28.self_attn.k_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.28.self_attn.o_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.28.self_attn.q_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.28.self_attn.rotary_emb.inv_freq": "model-00005-of-00015.safetensors",
+    "model.layers.28.self_attn.v_proj.weight": "model-00005-of-00015.safetensors",
+    "model.layers.29.input_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.29.mlp.down_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.29.mlp.gate_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.29.mlp.up_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.29.post_attention_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.29.self_attn.k_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.29.self_attn.o_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.29.self_attn.q_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.29.self_attn.rotary_emb.inv_freq": "model-00006-of-00015.safetensors",
+    "model.layers.29.self_attn.v_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.3.input_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.3.mlp.down_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.3.mlp.up_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.3.self_attn.rotary_emb.inv_freq": "model-00001-of-00015.safetensors",
+    "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.30.input_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.30.mlp.down_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.30.mlp.gate_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.30.mlp.up_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.30.post_attention_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.30.self_attn.k_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.30.self_attn.o_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.30.self_attn.q_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.30.self_attn.rotary_emb.inv_freq": "model-00006-of-00015.safetensors",
+    "model.layers.30.self_attn.v_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.input_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.mlp.down_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.mlp.gate_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.mlp.up_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.post_attention_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.self_attn.k_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.self_attn.o_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.self_attn.q_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.31.self_attn.rotary_emb.inv_freq": "model-00006-of-00015.safetensors",
+    "model.layers.31.self_attn.v_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.input_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.mlp.down_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.mlp.gate_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.mlp.up_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.post_attention_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.self_attn.k_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.self_attn.o_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.self_attn.q_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.32.self_attn.rotary_emb.inv_freq": "model-00006-of-00015.safetensors",
+    "model.layers.32.self_attn.v_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.input_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.mlp.down_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.mlp.gate_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.mlp.up_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.post_attention_layernorm.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.self_attn.k_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.self_attn.o_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.self_attn.q_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.33.self_attn.rotary_emb.inv_freq": "model-00006-of-00015.safetensors",
+    "model.layers.33.self_attn.v_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.34.input_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.34.mlp.down_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.34.mlp.gate_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.34.mlp.up_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.34.post_attention_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.34.self_attn.k_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.34.self_attn.o_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.34.self_attn.q_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.34.self_attn.rotary_emb.inv_freq": "model-00006-of-00015.safetensors",
+    "model.layers.34.self_attn.v_proj.weight": "model-00006-of-00015.safetensors",
+    "model.layers.35.input_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.35.mlp.down_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.35.mlp.gate_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.35.mlp.up_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.35.post_attention_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.35.self_attn.k_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.35.self_attn.o_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.35.self_attn.q_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.35.self_attn.rotary_emb.inv_freq": "model-00007-of-00015.safetensors",
+    "model.layers.35.self_attn.v_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.input_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.mlp.down_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.mlp.gate_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.mlp.up_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.post_attention_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.self_attn.k_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.self_attn.o_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.self_attn.q_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.36.self_attn.rotary_emb.inv_freq": "model-00007-of-00015.safetensors",
+    "model.layers.36.self_attn.v_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.input_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.mlp.down_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.mlp.gate_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.mlp.up_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.post_attention_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.self_attn.k_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.self_attn.o_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.self_attn.q_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.37.self_attn.rotary_emb.inv_freq": "model-00007-of-00015.safetensors",
+    "model.layers.37.self_attn.v_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.input_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.mlp.down_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.mlp.gate_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.mlp.up_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.post_attention_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.self_attn.k_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.self_attn.o_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.self_attn.q_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.38.self_attn.rotary_emb.inv_freq": "model-00007-of-00015.safetensors",
+    "model.layers.38.self_attn.v_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.input_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.mlp.down_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.mlp.gate_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.mlp.up_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.post_attention_layernorm.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.self_attn.k_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.self_attn.o_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.self_attn.q_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.39.self_attn.rotary_emb.inv_freq": "model-00007-of-00015.safetensors",
+    "model.layers.39.self_attn.v_proj.weight": "model-00007-of-00015.safetensors",
+    "model.layers.4.input_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.4.mlp.down_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.4.mlp.up_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00015.safetensors",
+    "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.4.self_attn.rotary_emb.inv_freq": "model-00001-of-00015.safetensors",
+    "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.40.input_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.40.mlp.down_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.40.mlp.gate_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.40.mlp.up_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.40.post_attention_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.40.self_attn.k_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.40.self_attn.o_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.40.self_attn.q_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.40.self_attn.rotary_emb.inv_freq": "model-00008-of-00015.safetensors",
+    "model.layers.40.self_attn.v_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.input_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.mlp.down_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.mlp.gate_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.mlp.up_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.post_attention_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.self_attn.k_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.self_attn.o_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.self_attn.q_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.41.self_attn.rotary_emb.inv_freq": "model-00008-of-00015.safetensors",
+    "model.layers.41.self_attn.v_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.input_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.mlp.down_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.mlp.gate_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.mlp.up_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.post_attention_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.self_attn.k_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.self_attn.o_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.self_attn.q_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.42.self_attn.rotary_emb.inv_freq": "model-00008-of-00015.safetensors",
+    "model.layers.42.self_attn.v_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.input_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.mlp.down_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.mlp.gate_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.mlp.up_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.post_attention_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.self_attn.k_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.self_attn.o_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.self_attn.q_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.43.self_attn.rotary_emb.inv_freq": "model-00008-of-00015.safetensors",
+    "model.layers.43.self_attn.v_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.input_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.mlp.down_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.mlp.gate_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.mlp.up_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.post_attention_layernorm.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.self_attn.k_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.self_attn.o_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.self_attn.q_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.44.self_attn.rotary_emb.inv_freq": "model-00008-of-00015.safetensors",
+    "model.layers.44.self_attn.v_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.45.input_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.45.mlp.down_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.45.mlp.gate_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.45.mlp.up_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.45.post_attention_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.45.self_attn.k_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.45.self_attn.o_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.45.self_attn.q_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.45.self_attn.rotary_emb.inv_freq": "model-00008-of-00015.safetensors",
+    "model.layers.45.self_attn.v_proj.weight": "model-00008-of-00015.safetensors",
+    "model.layers.46.input_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.46.mlp.down_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.46.mlp.gate_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.46.mlp.up_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.46.post_attention_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.46.self_attn.k_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.46.self_attn.o_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.46.self_attn.q_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.46.self_attn.rotary_emb.inv_freq": "model-00009-of-00015.safetensors",
+    "model.layers.46.self_attn.v_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.input_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.mlp.down_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.mlp.gate_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.mlp.up_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.post_attention_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.self_attn.k_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.self_attn.o_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.self_attn.q_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.47.self_attn.rotary_emb.inv_freq": "model-00009-of-00015.safetensors",
+    "model.layers.47.self_attn.v_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.input_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.mlp.down_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.mlp.gate_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.mlp.up_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.post_attention_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.self_attn.k_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.self_attn.o_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.self_attn.q_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.48.self_attn.rotary_emb.inv_freq": "model-00009-of-00015.safetensors",
+    "model.layers.48.self_attn.v_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.input_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.mlp.down_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.mlp.gate_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.mlp.up_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.post_attention_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.self_attn.k_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.self_attn.o_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.self_attn.q_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.49.self_attn.rotary_emb.inv_freq": "model-00009-of-00015.safetensors",
+    "model.layers.49.self_attn.v_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.5.input_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.5.mlp.down_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.5.mlp.up_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.5.self_attn.rotary_emb.inv_freq": "model-00001-of-00015.safetensors",
+    "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00015.safetensors",
+    "model.layers.50.input_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.50.mlp.down_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.50.mlp.gate_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.50.mlp.up_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.50.post_attention_layernorm.weight": "model-00009-of-00015.safetensors",
+    "model.layers.50.self_attn.k_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.50.self_attn.o_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.50.self_attn.q_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.50.self_attn.rotary_emb.inv_freq": "model-00009-of-00015.safetensors",
+    "model.layers.50.self_attn.v_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.51.input_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.51.mlp.down_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.51.mlp.gate_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.51.mlp.up_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.51.post_attention_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.51.self_attn.k_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.51.self_attn.o_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.51.self_attn.q_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.51.self_attn.rotary_emb.inv_freq": "model-00009-of-00015.safetensors",
+    "model.layers.51.self_attn.v_proj.weight": "model-00009-of-00015.safetensors",
+    "model.layers.52.input_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.52.mlp.down_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.52.mlp.gate_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.52.mlp.up_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.52.post_attention_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.52.self_attn.k_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.52.self_attn.o_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.52.self_attn.q_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.52.self_attn.rotary_emb.inv_freq": "model-00010-of-00015.safetensors",
+    "model.layers.52.self_attn.v_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.input_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.mlp.down_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.mlp.gate_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.mlp.up_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.post_attention_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.self_attn.k_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.self_attn.o_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.self_attn.q_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.53.self_attn.rotary_emb.inv_freq": "model-00010-of-00015.safetensors",
+    "model.layers.53.self_attn.v_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.input_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.mlp.down_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.mlp.gate_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.mlp.up_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.post_attention_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.self_attn.k_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.self_attn.o_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.self_attn.q_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.54.self_attn.rotary_emb.inv_freq": "model-00010-of-00015.safetensors",
+    "model.layers.54.self_attn.v_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.input_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.mlp.down_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.mlp.gate_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.mlp.up_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.post_attention_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.self_attn.k_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.self_attn.o_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.self_attn.q_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.55.self_attn.rotary_emb.inv_freq": "model-00010-of-00015.safetensors",
+    "model.layers.55.self_attn.v_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.input_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.mlp.down_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.mlp.gate_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.mlp.up_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.post_attention_layernorm.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.self_attn.k_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.self_attn.o_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.self_attn.q_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.56.self_attn.rotary_emb.inv_freq": "model-00010-of-00015.safetensors",
+    "model.layers.56.self_attn.v_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.57.input_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.57.mlp.down_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.57.mlp.gate_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.57.mlp.up_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.57.post_attention_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.57.self_attn.k_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.57.self_attn.o_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.57.self_attn.q_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.57.self_attn.rotary_emb.inv_freq": "model-00010-of-00015.safetensors",
+    "model.layers.57.self_attn.v_proj.weight": "model-00010-of-00015.safetensors",
+    "model.layers.58.input_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.58.mlp.down_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.58.mlp.gate_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.58.mlp.up_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.58.post_attention_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.58.self_attn.k_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.58.self_attn.o_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.58.self_attn.q_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.58.self_attn.rotary_emb.inv_freq": "model-00011-of-00015.safetensors",
+    "model.layers.58.self_attn.v_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.input_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.mlp.down_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.mlp.gate_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.mlp.up_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.post_attention_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.self_attn.k_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.self_attn.o_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.self_attn.q_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.59.self_attn.rotary_emb.inv_freq": "model-00011-of-00015.safetensors",
+    "model.layers.59.self_attn.v_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.6.input_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.6.mlp.down_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.6.mlp.gate_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.6.mlp.up_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.6.self_attn.rotary_emb.inv_freq": "model-00002-of-00015.safetensors",
+    "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.60.input_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.60.mlp.down_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.60.mlp.gate_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.60.mlp.up_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.60.post_attention_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.60.self_attn.k_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.60.self_attn.o_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.60.self_attn.q_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.60.self_attn.rotary_emb.inv_freq": "model-00011-of-00015.safetensors",
+    "model.layers.60.self_attn.v_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.input_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.mlp.down_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.mlp.gate_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.mlp.up_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.post_attention_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.self_attn.k_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.self_attn.o_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.self_attn.q_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.61.self_attn.rotary_emb.inv_freq": "model-00011-of-00015.safetensors",
+    "model.layers.61.self_attn.v_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.input_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.mlp.down_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.mlp.gate_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.mlp.up_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.post_attention_layernorm.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.self_attn.k_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.self_attn.o_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.self_attn.q_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.62.self_attn.rotary_emb.inv_freq": "model-00011-of-00015.safetensors",
+    "model.layers.62.self_attn.v_proj.weight": "model-00011-of-00015.safetensors",
+    "model.layers.63.input_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.63.mlp.down_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.63.mlp.gate_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.63.mlp.up_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.63.post_attention_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.63.self_attn.k_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.63.self_attn.o_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.63.self_attn.q_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.63.self_attn.rotary_emb.inv_freq": "model-00012-of-00015.safetensors",
+    "model.layers.63.self_attn.v_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.input_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.mlp.down_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.mlp.gate_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.mlp.up_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.post_attention_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.self_attn.k_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.self_attn.o_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.self_attn.q_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.64.self_attn.rotary_emb.inv_freq": "model-00012-of-00015.safetensors",
+    "model.layers.64.self_attn.v_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.input_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.mlp.down_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.mlp.gate_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.mlp.up_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.post_attention_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.self_attn.k_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.self_attn.o_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.self_attn.q_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.65.self_attn.rotary_emb.inv_freq": "model-00012-of-00015.safetensors",
+    "model.layers.65.self_attn.v_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.input_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.mlp.down_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.mlp.gate_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.mlp.up_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.post_attention_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.self_attn.k_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.self_attn.o_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.self_attn.q_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.66.self_attn.rotary_emb.inv_freq": "model-00012-of-00015.safetensors",
+    "model.layers.66.self_attn.v_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.input_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.mlp.down_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.mlp.gate_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.mlp.up_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.post_attention_layernorm.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.self_attn.k_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.self_attn.o_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.self_attn.q_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.67.self_attn.rotary_emb.inv_freq": "model-00012-of-00015.safetensors",
+    "model.layers.67.self_attn.v_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.68.input_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.68.mlp.down_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.68.mlp.gate_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.68.mlp.up_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.68.post_attention_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.68.self_attn.k_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.68.self_attn.o_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.68.self_attn.q_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.68.self_attn.rotary_emb.inv_freq": "model-00012-of-00015.safetensors",
+    "model.layers.68.self_attn.v_proj.weight": "model-00012-of-00015.safetensors",
+    "model.layers.69.input_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.69.mlp.down_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.69.mlp.gate_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.69.mlp.up_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.69.post_attention_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.69.self_attn.k_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.69.self_attn.o_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.69.self_attn.q_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.69.self_attn.rotary_emb.inv_freq": "model-00013-of-00015.safetensors",
+    "model.layers.69.self_attn.v_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.7.input_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.7.mlp.down_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.7.mlp.gate_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.7.mlp.up_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.7.self_attn.k_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.7.self_attn.o_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.7.self_attn.q_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.7.self_attn.rotary_emb.inv_freq": "model-00002-of-00015.safetensors",
+    "model.layers.7.self_attn.v_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.70.input_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.70.mlp.down_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.70.mlp.gate_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.70.mlp.up_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.70.post_attention_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.70.self_attn.k_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.70.self_attn.o_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.70.self_attn.q_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.70.self_attn.rotary_emb.inv_freq": "model-00013-of-00015.safetensors",
+    "model.layers.70.self_attn.v_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.input_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.mlp.down_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.mlp.gate_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.mlp.up_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.post_attention_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.self_attn.k_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.self_attn.o_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.self_attn.q_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.71.self_attn.rotary_emb.inv_freq": "model-00013-of-00015.safetensors",
+    "model.layers.71.self_attn.v_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.input_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.mlp.down_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.mlp.gate_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.mlp.up_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.post_attention_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.self_attn.k_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.self_attn.o_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.self_attn.q_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.72.self_attn.rotary_emb.inv_freq": "model-00013-of-00015.safetensors",
+    "model.layers.72.self_attn.v_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.input_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.mlp.down_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.mlp.gate_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.mlp.up_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.post_attention_layernorm.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.self_attn.k_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.self_attn.o_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.self_attn.q_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.73.self_attn.rotary_emb.inv_freq": "model-00013-of-00015.safetensors",
+    "model.layers.73.self_attn.v_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.74.input_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.74.mlp.down_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.74.mlp.gate_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.74.mlp.up_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.74.post_attention_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.74.self_attn.k_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.74.self_attn.o_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.74.self_attn.q_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.74.self_attn.rotary_emb.inv_freq": "model-00013-of-00015.safetensors",
+    "model.layers.74.self_attn.v_proj.weight": "model-00013-of-00015.safetensors",
+    "model.layers.75.input_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.75.mlp.down_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.75.mlp.gate_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.75.mlp.up_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.75.post_attention_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.75.self_attn.k_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.75.self_attn.o_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.75.self_attn.q_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.75.self_attn.rotary_emb.inv_freq": "model-00014-of-00015.safetensors",
+    "model.layers.75.self_attn.v_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.input_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.mlp.down_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.mlp.gate_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.mlp.up_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.post_attention_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.self_attn.k_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.self_attn.o_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.self_attn.q_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.76.self_attn.rotary_emb.inv_freq": "model-00014-of-00015.safetensors",
+    "model.layers.76.self_attn.v_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.input_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.mlp.down_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.mlp.gate_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.mlp.up_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.post_attention_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.self_attn.k_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.self_attn.o_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.self_attn.q_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.77.self_attn.rotary_emb.inv_freq": "model-00014-of-00015.safetensors",
+    "model.layers.77.self_attn.v_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.input_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.mlp.down_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.mlp.gate_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.mlp.up_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.post_attention_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.self_attn.k_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.self_attn.o_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.self_attn.q_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.78.self_attn.rotary_emb.inv_freq": "model-00014-of-00015.safetensors",
+    "model.layers.78.self_attn.v_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.input_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.mlp.down_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.mlp.gate_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.mlp.up_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.post_attention_layernorm.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.self_attn.k_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.self_attn.o_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.self_attn.q_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.79.self_attn.rotary_emb.inv_freq": "model-00014-of-00015.safetensors",
+    "model.layers.79.self_attn.v_proj.weight": "model-00014-of-00015.safetensors",
+    "model.layers.8.input_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.8.mlp.down_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.8.mlp.gate_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.8.mlp.up_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.8.self_attn.rotary_emb.inv_freq": "model-00002-of-00015.safetensors",
+    "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.input_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.mlp.down_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.mlp.gate_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.mlp.up_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00015.safetensors",
+    "model.layers.9.self_attn.rotary_emb.inv_freq": "model-00002-of-00015.safetensors",
+    "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00015.safetensors",
+    "model.norm.weight": "model-00014-of-00015.safetensors"
+  }
+}

modeling_dropped_llama.py

@@ -0,0 +1,1338 @@
+# coding=utf-8
+# Copyright 2022 EleutherAI and the HuggingFace Inc. team. All rights reserved.
+#
+# This code is based on EleutherAI's GPT-NeoX library and the GPT-NeoX
+# and OPT implementations in this library. It has been modified from its
+# original forms to accommodate minor architectural differences compared
+# to GPT-NeoX and OPT used by the Meta AI team that trained the model.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+""" transformers==4.38.1"""
+""" PyTorch LLaMA model."""
+import math
+import warnings
+from typing import List, Optional, Tuple, Union
+
+import torch
+import torch.nn.functional as F
+import torch.utils.checkpoint
+from torch import nn
+from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
+
+from transformers.activations import ACT2FN
+from transformers.cache_utils import Cache, DynamicCache, StaticCache
+from transformers.modeling_outputs import (
+    BaseModelOutputWithPast,
+    CausalLMOutputWithPast,
+    QuestionAnsweringModelOutput,
+    SequenceClassifierOutputWithPast,
+)
+from transformers.modeling_utils import PreTrainedModel
+from transformers.pytorch_utils import ALL_LAYERNORM_LAYERS
+from transformers.utils import (
+    add_start_docstrings,
+    add_start_docstrings_to_model_forward,
+    is_flash_attn_2_available,
+    is_flash_attn_greater_or_equal_2_10,
+    logging,
+    replace_return_docstrings,
+)
+from .configuration_dropped_llama import LlamaConfig
+
+
+# if is_flash_attn_2_available():
+#     from flash_attn import flash_attn_func, flash_attn_varlen_func
+#     from flash_attn.bert_padding import index_first_axis, pad_input, unpad_input  # noqa
+
+
+logger = logging.get_logger(__name__)
+
+_CONFIG_FOR_DOC = "LlamaConfig"
+
+
+def _get_unpad_data(attention_mask):
+    seqlens_in_batch = attention_mask.sum(dim=-1, dtype=torch.int32)
+    indices = torch.nonzero(attention_mask.flatten(), as_tuple=False).flatten()
+    max_seqlen_in_batch = seqlens_in_batch.max().item()
+    cu_seqlens = F.pad(torch.cumsum(seqlens_in_batch, dim=0, dtype=torch.int32), (1, 0))
+    return (
+        indices,
+        cu_seqlens,
+        max_seqlen_in_batch,
+    )
+
+
+class LlamaRMSNorm(nn.Module):
+    def __init__(self, hidden_size, eps=1e-6):
+        """
+        LlamaRMSNorm is equivalent to T5LayerNorm
+        """
+        super().__init__()
+        self.weight = nn.Parameter(torch.ones(hidden_size))
+        self.variance_epsilon = eps
+
+    def forward(self, hidden_states):
+        input_dtype = hidden_states.dtype
+        hidden_states = hidden_states.to(torch.float32)
+        variance = hidden_states.pow(2).mean(-1, keepdim=True)
+        hidden_states = hidden_states * torch.rsqrt(variance + self.variance_epsilon)
+        return self.weight * hidden_states.to(input_dtype)
+
+
+ALL_LAYERNORM_LAYERS.append(LlamaRMSNorm)
+
+
+class LlamaRotaryEmbedding(nn.Module):
+    def __init__(self, dim, max_position_embeddings=2048, base=10000, device=None):
+        super().__init__()
+        self.dim = dim
+        self.max_position_embeddings = max_position_embeddings
+        self.base = base
+        inv_freq = 1.0 / (self.base ** (torch.arange(0, self.dim, 2, dtype=torch.int64).float().to(device) / self.dim))
+        self.register_buffer("inv_freq", inv_freq, persistent=False)
+
+    @property
+    def sin_cached(self):
+        logger.warning_once(
+            "The sin_cached attribute will be removed in 4.40. Bear in mind that its contents changed in v4.38. Use "
+            "the forward method of RoPE from now on instead."
+        )
+        return self._sin_cached
+
+    @property
+    def cos_cached(self):
+        logger.warning_once(
+            "The cos_cached attribute will be removed in 4.40. Bear in mind that its contents changed in v4.38. Use "
+            "the forward method of RoPE from now on instead."
+        )
+        return self._cos_cached
+
+    def forward(self, x, position_ids, seq_len=None):
+        if seq_len is not None:
+            logger.warning_once("The `seq_len` argument is deprecated and unused. It will be removed in v4.40.")
+
+        # x: [bs, num_attention_heads, seq_len, head_size]
+        inv_freq_expanded = self.inv_freq[None, :, None].float().expand(position_ids.shape[0], -1, 1)
+        position_ids_expanded = position_ids[:, None, :].float()
+        freqs = (inv_freq_expanded @ position_ids_expanded).transpose(1, 2)
+        emb = torch.cat((freqs, freqs), dim=-1)
+        cos = emb.cos().to(dtype=x.dtype)
+        sin = emb.sin().to(dtype=x.dtype)
+        # backwards compatibility
+        self._cos_cached = cos
+        self._sin_cached = sin
+        return cos, sin
+
+
+class LlamaLinearScalingRotaryEmbedding(LlamaRotaryEmbedding):
+    """LlamaRotaryEmbedding extended with linear scaling. Credits to the Reddit user /u/kaiokendev"""
+
+    def __init__(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0):
+        self.scaling_factor = scaling_factor
+        super().__init__(dim, max_position_embeddings, base, device)
+
+    def forward(self, x, position_ids, seq_len=None):
+        # difference to the original RoPE: a scaling factor is aplied to the position ids
+        position_ids = position_ids.float() / self.scaling_factor
+        cos, sin = super().forward(x, position_ids, seq_len)
+        return cos, sin
+
+
+class LlamaDynamicNTKScalingRotaryEmbedding(LlamaRotaryEmbedding):
+    """LlamaRotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla"""
+
+    def __init__(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0):
+        self.scaling_factor = scaling_factor
+        super().__init__(dim, max_position_embeddings, base, device)
+
+    def forward(self, x, position_ids, seq_len=None):
+        # difference to the original RoPE: inv_freq is recomputed when the sequence length > original length
+        seq_len = torch.max(position_ids) + 1
+        if seq_len > self.max_position_embeddings:
+            base = self.base * (
+                (self.scaling_factor * seq_len / self.max_position_embeddings) - (self.scaling_factor - 1)
+            ) ** (self.dim / (self.dim - 2))
+            inv_freq = 1.0 / (
+                base ** (torch.arange(0, self.dim, 2, dtype=torch.int64).float().to(x.device) / self.dim)
+            )
+            self.register_buffer("inv_freq", inv_freq, persistent=False)  # TODO joao: this may break with compilation
+
+        cos, sin = super().forward(x, position_ids, seq_len)
+        return cos, sin
+
+
+def rotate_half(x):
+    """Rotates half the hidden dims of the input."""
+    x1 = x[..., : x.shape[-1] // 2]
+    x2 = x[..., x.shape[-1] // 2 :]
+    return torch.cat((-x2, x1), dim=-1)
+
+
+def apply_rotary_pos_emb(q, k, cos, sin, position_ids=None, unsqueeze_dim=1):
+    """Applies Rotary Position Embedding to the query and key tensors.
+
+    Args:
+        q (`torch.Tensor`): The query tensor.
+        k (`torch.Tensor`): The key tensor.
+        cos (`torch.Tensor`): The cosine part of the rotary embedding.
+        sin (`torch.Tensor`): The sine part of the rotary embedding.
+        position_ids (`torch.Tensor`, *optional*):
+            Deprecated and unused.
+        unsqueeze_dim (`int`, *optional*, defaults to 1):
+            The 'unsqueeze_dim' argument specifies the dimension along which to unsqueeze cos[position_ids] and
+            sin[position_ids] so that they can be properly broadcasted to the dimensions of q and k. For example, note
+            that cos[position_ids] and sin[position_ids] have the shape [batch_size, seq_len, head_dim]. Then, if q and
+            k have the shape [batch_size, heads, seq_len, head_dim], then setting unsqueeze_dim=1 makes
+            cos[position_ids] and sin[position_ids] broadcastable to the shapes of q and k. Similarly, if q and k have
+            the shape [batch_size, seq_len, heads, head_dim], then set unsqueeze_dim=2.
+    Returns:
+        `tuple(torch.Tensor)` comprising of the query and key tensors rotated using the Rotary Position Embedding.
+    """
+    cos = cos.unsqueeze(unsqueeze_dim)
+    sin = sin.unsqueeze(unsqueeze_dim)
+    q_embed = (q * cos) + (rotate_half(q) * sin)
+    k_embed = (k * cos) + (rotate_half(k) * sin)
+    return q_embed, k_embed
+
+
+class LlamaMLP(nn.Module):
+    def __init__(self, config):
+        super().__init__()
+        self.config = config
+        self.hidden_size = config.hidden_size
+        self.intermediate_size = config.intermediate_size
+        self.gate_proj = nn.Linear(self.hidden_size, self.intermediate_size, bias=False)
+        self.up_proj = nn.Linear(self.hidden_size, self.intermediate_size, bias=False)
+        self.down_proj = nn.Linear(self.intermediate_size, self.hidden_size, bias=False)
+        self.act_fn = ACT2FN[config.hidden_act]
+
+    def forward(self, x):
+        if self.config.pretraining_tp > 1:
+            slice = self.intermediate_size // self.config.pretraining_tp
+            gate_proj_slices = self.gate_proj.weight.split(slice, dim=0)
+            up_proj_slices = self.up_proj.weight.split(slice, dim=0)
+            down_proj_slices = self.down_proj.weight.split(slice, dim=1)
+
+            gate_proj = torch.cat(
+                [F.linear(x, gate_proj_slices[i]) for i in range(self.config.pretraining_tp)], dim=-1
+            )
+            up_proj = torch.cat([F.linear(x, up_proj_slices[i]) for i in range(self.config.pretraining_tp)], dim=-1)
+
+            intermediate_states = (self.act_fn(gate_proj) * up_proj).split(slice, dim=2)
+            down_proj = [
+                F.linear(intermediate_states[i], down_proj_slices[i]) for i in range(self.config.pretraining_tp)
+            ]
+            down_proj = sum(down_proj)
+        else:
+            down_proj = self.down_proj(self.act_fn(self.gate_proj(x)) * self.up_proj(x))
+
+        return down_proj
+
+
+def repeat_kv(hidden_states: torch.Tensor, n_rep: int) -> torch.Tensor:
+    """
+    This is the equivalent of torch.repeat_interleave(x, dim=1, repeats=n_rep). The hidden states go from (batch,
+    num_key_value_heads, seqlen, head_dim) to (batch, num_attention_heads, seqlen, head_dim)
+    """
+    batch, num_key_value_heads, slen, head_dim = hidden_states.shape
+    if n_rep == 1:
+        return hidden_states
+    hidden_states = hidden_states[:, :, None, :, :].expand(batch, num_key_value_heads, n_rep, slen, head_dim)
+    return hidden_states.reshape(batch, num_key_value_heads * n_rep, slen, head_dim)
+
+
+class LlamaAttention(nn.Module):
+    """Multi-headed attention from 'Attention Is All You Need' paper"""
+
+    def __init__(self, config: LlamaConfig, layer_idx: Optional[int] = None, kv_cache_idx: Optional[int] = None):
+        super().__init__()
+        self.config = config
+        self.layer_idx = layer_idx
+        self.kv_cache_idx = kv_cache_idx
+        if layer_idx is None:
+            logger.warning_once(
+                f"Instantiating {self.__class__.__name__} without passing a `layer_idx` is not recommended and will "
+                "lead to errors during the forward call if caching is used. Please make sure to provide a `layer_idx` "
+                "when creating this class."
+            )
+
+        self.attention_dropout = config.attention_dropout
+        self.hidden_size = config.hidden_size
+        self.num_heads = config.num_attention_heads
+        self.head_dim = self.hidden_size // self.num_heads
+        self.num_key_value_heads = config.num_key_value_heads
+        self.num_key_value_groups = self.num_heads // self.num_key_value_heads
+        self.max_position_embeddings = config.max_position_embeddings
+        self.rope_theta = config.rope_theta
+        self.is_causal = True
+
+        if (self.head_dim * self.num_heads) != self.hidden_size:
+            raise ValueError(
+                f"hidden_size must be divisible by num_heads (got `hidden_size`: {self.hidden_size}"
+                f" and `num_heads`: {self.num_heads})."
+            )
+
+        self.q_proj = nn.Linear(self.hidden_size, self.num_heads * self.head_dim, bias=config.attention_bias)
+        self.k_proj = nn.Linear(self.hidden_size, self.num_key_value_heads * self.head_dim, bias=config.attention_bias)
+        self.v_proj = nn.Linear(self.hidden_size, self.num_key_value_heads * self.head_dim, bias=config.attention_bias)
+        self.o_proj = nn.Linear(self.hidden_size, self.hidden_size, bias=config.attention_bias)
+        self._init_rope()
+
+    def _init_rope(self):
+        if self.config.rope_scaling is None:
+            self.rotary_emb = LlamaRotaryEmbedding(
+                self.head_dim,
+                max_position_embeddings=self.max_position_embeddings,
+                base=self.rope_theta,
+            )
+        else:
+            scaling_type = self.config.rope_scaling["type"]
+            scaling_factor = self.config.rope_scaling["factor"]
+            if scaling_type == "linear":
+                self.rotary_emb = LlamaLinearScalingRotaryEmbedding(
+                    self.head_dim,
+                    max_position_embeddings=self.max_position_embeddings,
+                    scaling_factor=scaling_factor,
+                    base=self.rope_theta,
+                )
+            elif scaling_type == "dynamic":
+                self.rotary_emb = LlamaDynamicNTKScalingRotaryEmbedding(
+                    self.head_dim,
+                    max_position_embeddings=self.max_position_embeddings,
+                    scaling_factor=scaling_factor,
+                    base=self.rope_theta,
+                )
+            else:
+                raise ValueError(f"Unknown RoPE scaling type {scaling_type}")
+
+    def forward(
+        self,
+        hidden_states: torch.Tensor,
+        attention_mask: Optional[torch.Tensor] = None,
+        position_ids: Optional[torch.LongTensor] = None,
+        past_key_value: Optional[Cache] = None,
+        output_attentions: bool = False,
+        use_cache: bool = False,
+        cache_position: Optional[torch.LongTensor] = None,
+        **kwargs,
+    ) -> Tuple[torch.Tensor, Optional[torch.Tensor], Optional[Tuple[torch.Tensor]]]:
+        bsz, q_len, _ = hidden_states.size()
+
+        if self.config.pretraining_tp > 1:
+            key_value_slicing = (self.num_key_value_heads * self.head_dim) // self.config.pretraining_tp
+            query_slices = self.q_proj.weight.split(
+                (self.num_heads * self.head_dim) // self.config.pretraining_tp, dim=0
+            )
+            key_slices = self.k_proj.weight.split(key_value_slicing, dim=0)
+            value_slices = self.v_proj.weight.split(key_value_slicing, dim=0)
+
+            query_states = [F.linear(hidden_states, query_slices[i]) for i in range(self.config.pretraining_tp)]
+            query_states = torch.cat(query_states, dim=-1)
+
+            key_states = [F.linear(hidden_states, key_slices[i]) for i in range(self.config.pretraining_tp)]
+            key_states = torch.cat(key_states, dim=-1)
+
+            value_states = [F.linear(hidden_states, value_slices[i]) for i in range(self.config.pretraining_tp)]
+            value_states = torch.cat(value_states, dim=-1)
+
+        else:
+            query_states = self.q_proj(hidden_states)
+            key_states = self.k_proj(hidden_states)
+            value_states = self.v_proj(hidden_states)
+
+        query_states = query_states.view(bsz, q_len, self.num_heads, self.head_dim).transpose(1, 2)
+        key_states = key_states.view(bsz, q_len, self.num_key_value_heads, self.head_dim).transpose(1, 2)
+        value_states = value_states.view(bsz, q_len, self.num_key_value_heads, self.head_dim).transpose(1, 2)
+
+        past_key_value = getattr(self, "past_key_value", past_key_value)
+        cos, sin = self.rotary_emb(value_states, position_ids)
+        query_states, key_states = apply_rotary_pos_emb(query_states, key_states, cos, sin)
+
+        if past_key_value is not None:
+            # sin and cos are specific to RoPE models; position_ids needed for the static cache
+            cache_kwargs = {"sin": sin, "cos": cos, "cache_position": cache_position}
+            key_states, value_states = past_key_value.update(key_states, value_states, self.kv_cache_idx, cache_kwargs)
+
+        key_states = repeat_kv(key_states, self.num_key_value_groups)
+        value_states = repeat_kv(value_states, self.num_key_value_groups)
+
+        attn_weights = torch.matmul(query_states, key_states.transpose(2, 3)) / math.sqrt(self.head_dim)
+
+        if attention_mask is not None:  # no matter the length, we just slice it
+            if cache_position is not None:
+                causal_mask = attention_mask[:, :, cache_position, : key_states.shape[-2]]
+            attn_weights = attn_weights + causal_mask
+
+        # upcast attention to fp32
+        attn_weights = nn.functional.softmax(attn_weights, dim=-1, dtype=torch.float32).to(query_states.dtype)
+        attn_weights = nn.functional.dropout(attn_weights, p=self.attention_dropout, training=self.training)
+        attn_output = torch.matmul(attn_weights, value_states)
+
+        if attn_output.size() != (bsz, self.num_heads, q_len, self.head_dim):
+            raise ValueError(
+                f"`attn_output` should be of size {(bsz, self.num_heads, q_len, self.head_dim)}, but is"
+                f" {attn_output.size()}"
+            )
+
+        attn_output = attn_output.transpose(1, 2).contiguous()
+
+        attn_output = attn_output.reshape(bsz, q_len, self.hidden_size)
+
+        if self.config.pretraining_tp > 1:
+            attn_output = attn_output.split(self.hidden_size // self.config.pretraining_tp, dim=2)
+            o_proj_slices = self.o_proj.weight.split(self.hidden_size // self.config.pretraining_tp, dim=1)
+            attn_output = sum([F.linear(attn_output[i], o_proj_slices[i]) for i in range(self.config.pretraining_tp)])
+        else:
+            attn_output = self.o_proj(attn_output)
+
+        if not output_attentions:
+            attn_weights = None
+
+        return attn_output, attn_weights, past_key_value
+
+
+class LlamaSdpaAttention(LlamaAttention):
+    """
+    Llama attention module using torch.nn.functional.scaled_dot_product_attention. This module inherits from
+    `LlamaAttention` as the weights of the module stays untouched. The only changes are on the forward pass to adapt to
+    SDPA API.
+    """
+
+    # Adapted from LlamaAttention.forward
+    def forward(
+        self,
+        hidden_states: torch.Tensor,
+        attention_mask: Optional[torch.Tensor] = None,
+        position_ids: Optional[torch.LongTensor] = None,
+        past_key_value: Optional[Cache] = None,
+        output_attentions: bool = False,
+        use_cache: bool = False,
+        cache_position: Optional[torch.LongTensor] = None,
+    ) -> Tuple[torch.Tensor, Optional[torch.Tensor], Optional[Tuple[torch.Tensor]]]:
+        if output_attentions:
+            # TODO: Improve this warning with e.g. `model.config.attn_implementation = "manual"` once this is implemented.
+            logger.warning_once(
+                "LlamaModel is using LlamaSdpaAttention, but `torch.nn.functional.scaled_dot_product_attention` does not support `output_attentions=True`. Falling back to the manual attention implementation, "
+                'but specifying the manual implementation will be required from Transformers version v5.0.0 onwards. This warning can be removed using the argument `attn_implementation="eager"` when loading the model.'
+            )
+            return super().forward(
+                hidden_states=hidden_states,
+                attention_mask=attention_mask,
+                position_ids=position_ids,
+                past_key_value=past_key_value,
+                output_attentions=output_attentions,
+                use_cache=use_cache,
+                cache_position=cache_position,
+            )
+
+        bsz, q_len, _ = hidden_states.size()
+
+        query_states = self.q_proj(hidden_states)
+        key_states = self.k_proj(hidden_states)
+        value_states = self.v_proj(hidden_states)
+
+        query_states = query_states.view(bsz, q_len, self.num_heads, self.head_dim).transpose(1, 2)
+        key_states = key_states.view(bsz, q_len, self.num_key_value_heads, self.head_dim).transpose(1, 2)
+        value_states = value_states.view(bsz, q_len, self.num_key_value_heads, self.head_dim).transpose(1, 2)
+
+        cos, sin = self.rotary_emb(value_states, position_ids)
+        query_states, key_states = apply_rotary_pos_emb(query_states, key_states, cos, sin)
+
+        past_key_value = getattr(self, "past_key_value", past_key_value)
+
+        if past_key_value is not None:
+            # sin and cos are specific to RoPE models; position_ids needed for the static cache
+            cache_kwargs = {"sin": sin, "cos": cos, "cache_position": cache_position}
+            key_states, value_states = past_key_value.update(key_states, value_states, self.kv_cache_idx, cache_kwargs)
+
+        key_states = repeat_kv(key_states, self.num_key_value_groups)
+        value_states = repeat_kv(value_states, self.num_key_value_groups)
+
+        causal_mask = attention_mask
+        if attention_mask is not None and cache_position is not None:
+            causal_mask = causal_mask[:, :, cache_position, : key_states.shape[-2]]
+
+        # SDPA with memory-efficient backend is currently (torch==2.1.2) bugged with non-contiguous inputs with custom attn_mask,
+        # Reference: https://github.com/pytorch/pytorch/issues/112577.
+        if query_states.device.type == "cuda" and causal_mask is not None:
+            query_states = query_states.contiguous()
+            key_states = key_states.contiguous()
+            value_states = value_states.contiguous()
+
+        attn_output = torch.nn.functional.scaled_dot_product_attention(
+            query_states,
+            key_states,
+            value_states,
+            attn_mask=causal_mask,
+            dropout_p=self.attention_dropout if self.training else 0.0,
+        )
+
+        attn_output = attn_output.transpose(1, 2).contiguous()
+        attn_output = attn_output.view(bsz, q_len, self.hidden_size)
+
+        attn_output = self.o_proj(attn_output)
+
+        return attn_output, None, past_key_value
+
+
+LLAMA_ATTENTION_CLASSES = {
+    "eager": LlamaAttention,
+    "sdpa": LlamaSdpaAttention,
+}
+
+
+class LlamaDecoderLayer(nn.Module):
+    def __init__(self, config: LlamaConfig, layer_idx: int):
+        super().__init__()
+        self.hidden_size = config.hidden_size
+        self.layer_idx = layer_idx
+        
+        self.kv_cache_idx = 0
+        for i in range(self.layer_idx):
+            if not config.drop_attn_list[i]:
+                self.kv_cache_idx += 1
+
+        self.drop_attn = config.drop_attn_list[layer_idx]
+        if self.drop_attn:
+            self.self_attn = None
+            self.input_layernorm = None
+        else:
+            self.self_attn = LLAMA_ATTENTION_CLASSES[config._attn_implementation](config=config, layer_idx=layer_idx, kv_cache_idx=self.kv_cache_idx)
+            self.input_layernorm = LlamaRMSNorm(config.hidden_size, eps=config.rms_norm_eps)
+        self.drop_mlp = config.drop_mlp_list[layer_idx]
+        if self.drop_mlp:
+            self.mlp = None
+            self.post_attention_layernorm = None
+        else:
+            self.mlp = LlamaMLP(config)
+            self.post_attention_layernorm = LlamaRMSNorm(config.hidden_size, eps=config.rms_norm_eps)
+
+
+    def forward(
+        self,
+        hidden_states: torch.Tensor,
+        attention_mask: Optional[torch.Tensor] = None,
+        position_ids: Optional[torch.LongTensor] = None,
+        past_key_value: Optional[Tuple[torch.Tensor]] = None,
+        output_attentions: Optional[bool] = False,
+        use_cache: Optional[bool] = False,
+        cache_position: Optional[torch.LongTensor] = None,
+        **kwargs,
+    ) -> Tuple[torch.FloatTensor, Optional[Tuple[torch.FloatTensor, torch.FloatTensor]]]:
+        """
+        Args:
+            hidden_states (`torch.FloatTensor`): input to the layer of shape `(batch, seq_len, embed_dim)`
+            attention_mask (`torch.FloatTensor`, *optional*):
+                attention mask of size `(batch_size, sequence_length)` if flash attention is used or `(batch_size, 1,
+                query_sequence_length, key_sequence_length)` if default attention is used.
+            output_attentions (`bool`, *optional*):
+                Whether or not to return the attentions tensors of all attention layers. See `attentions` under
+                returned tensors for more detail.
+            use_cache (`bool`, *optional*):
+                If set to `True`, `past_key_values` key value states are returned and can be used to speed up decoding
+                (see `past_key_values`).
+            past_key_value (`Tuple(torch.FloatTensor)`, *optional*): cached past key and value projection states
+        """
+        if "padding_mask" in kwargs:
+            warnings.warn(
+                "Passing `padding_mask` is deprecated and will be removed in v4.37. Please make sure use `attention_mask` instead.`"
+            )
+
+        if not self.drop_attn:
+            residual = hidden_states
+
+            hidden_states = self.input_layernorm(hidden_states)
+
+            # Self Attention
+            hidden_states, self_attn_weights, present_key_value = self.self_attn(
+                hidden_states=hidden_states,
+                attention_mask=attention_mask,
+                position_ids=position_ids,
+                past_key_value=past_key_value,
+                output_attentions=output_attentions,
+                use_cache=use_cache,
+                cache_position=cache_position,
+                **kwargs,
+            )
+            hidden_states = residual + hidden_states
+        
+        if not self.drop_mlp:
+            # Fully Connected
+            residual = hidden_states
+            hidden_states = self.post_attention_layernorm(hidden_states)
+            hidden_states = self.mlp(hidden_states)
+            hidden_states = residual + hidden_states
+
+        outputs = (hidden_states,)
+
+        if output_attentions:
+            outputs += (self_attn_weights,)
+        if use_cache and not self.drop_attn:
+            outputs += (present_key_value,)
+        # print(outputs)
+        return outputs
+
+
+LLAMA_START_DOCSTRING = r"""
+    This model inherits from [`PreTrainedModel`]. Check the superclass documentation for the generic methods the
+    library implements for all its model (such as downloading or saving, resizing the input embeddings, pruning heads
+    etc.)
+
+    This model is also a PyTorch [torch.nn.Module](https://pytorch.org/docs/stable/nn.html#torch.nn.Module) subclass.
+    Use it as a regular PyTorch Module and refer to the PyTorch documentation for all matter related to general usage
+    and behavior.
+
+    Parameters:
+        config ([`LlamaConfig`]):
+            Model configuration class with all the parameters of the model. Initializing with a config file does not
+            load the weights associated with the model, only the configuration. Check out the
+            [`~PreTrainedModel.from_pretrained`] method to load the model weights.
+"""
+
+
+@add_start_docstrings(
+    "The bare LLaMA Model outputting raw hidden-states without any specific head on top.",
+    LLAMA_START_DOCSTRING,
+)
+class LlamaPreTrainedModel(PreTrainedModel):
+    config_class = LlamaConfig
+    base_model_prefix = "model"
+    supports_gradient_checkpointing = True
+    _no_split_modules = ["LlamaDecoderLayer"]
+    _skip_keys_device_placement = ["past_key_values", "causal_mask"]
+    _supports_flash_attn_2 = True
+    _supports_sdpa = True
+    _supports_cache_class = True
+
+    def _init_weights(self, module):
+        std = self.config.initializer_range
+        if isinstance(module, nn.Linear):
+            module.weight.data.normal_(mean=0.0, std=std)
+            if module.bias is not None:
+                module.bias.data.zero_()
+        elif isinstance(module, nn.Embedding):
+            module.weight.data.normal_(mean=0.0, std=std)
+            if module.padding_idx is not None:
+                module.weight.data[module.padding_idx].zero_()
+
+    def _setup_cache(self, cache_cls, max_batch_size, max_cache_len: Optional[int] = None):
+        if self.config._attn_implementation == "flash_attention_2" and cache_cls == StaticCache:
+            raise ValueError(
+                "`static` cache implementation is not compatible with `attn_implementation==flash_attention_2` "
+                "make sure to use `sdpa` in the mean time, and open an issue at https://github.com/huggingface/transformers"
+            )
+
+        if max_cache_len > self.model.causal_mask.shape[-1] or self.device != self.model.causal_mask.device:
+            causal_mask = torch.full((max_cache_len, max_cache_len), fill_value=1, device=self.device)
+            self.register_buffer("causal_mask", torch.triu(causal_mask, diagonal=1), persistent=False)
+
+        for layer in self.model.layers:
+            weights = layer.self_attn.o_proj.weight
+            layer.self_attn.past_key_value = cache_cls(
+                self.config, max_batch_size, max_cache_len, device=weights.device, dtype=weights.dtype
+            )
+
+    def _reset_cache(self):
+        for layer in self.model.layers:
+            layer.self_attn.past_key_value = None
+
+
+LLAMA_INPUTS_DOCSTRING = r"""
+    Args:
+        input_ids (`torch.LongTensor` of shape `(batch_size, sequence_length)`):
+            Indices of input sequence tokens in the vocabulary. Padding will be ignored by default should you provide
+            it.
+
+            Indices can be obtained using [`AutoTokenizer`]. See [`PreTrainedTokenizer.encode`] and
+            [`PreTrainedTokenizer.__call__`] for details.
+
+            [What are input IDs?](../glossary#input-ids)
+        attention_mask (`torch.Tensor` of shape `(batch_size, sequence_length)`, *optional*):
+            Mask to avoid performing attention on padding token indices. Mask values selected in `[0, 1]`:
+
+            - 1 for tokens that are **not masked**,
+            - 0 for tokens that are **masked**.
+
+            [What are attention masks?](../glossary#attention-mask)
+
+            Indices can be obtained using [`AutoTokenizer`]. See [`PreTrainedTokenizer.encode`] and
+            [`PreTrainedTokenizer.__call__`] for details.
+
+            If `past_key_values` is used, optionally only the last `input_ids` have to be input (see
+            `past_key_values`).
+
+            If you want to change padding behavior, you should read [`modeling_opt._prepare_decoder_attention_mask`]
+            and modify to your needs. See diagram 1 in [the paper](https://arxiv.org/abs/1910.13461) for more
+            information on the default strategy.
+
+            - 1 indicates the head is **not masked**,
+            - 0 indicates the head is **masked**.
+        position_ids (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*):
+            Indices of positions of each input sequence tokens in the position embeddings. Selected in the range `[0,
+            config.n_positions - 1]`.
+
+            [What are position IDs?](../glossary#position-ids)
+        past_key_values (`Cache` or `tuple(tuple(torch.FloatTensor))`, *optional*):
+            Pre-computed hidden-states (key and values in the self-attention blocks and in the cross-attention
+            blocks) that can be used to speed up sequential decoding. This typically consists in the `past_key_values`
+            returned by the model at a previous stage of decoding, when `use_cache=True` or `config.use_cache=True`.
+
+            Two formats are allowed:
+            - a [`~cache_utils.Cache`] instance;
+            - Tuple of `tuple(torch.FloatTensor)` of length `config.n_layers`, with each tuple having 2 tensors of
+            shape `(batch_size, num_heads, sequence_length, embed_size_per_head)`). This is also known as the legacy
+            cache format.
+
+            The model will output the same cache format that is fed as input. If no `past_key_values` are passed, the
+            legacy cache format will be returned.
+
+            If `past_key_values` are used, the user can optionally input only the last `input_ids` (those that don't
+            have their past key value states given to this model) of shape `(batch_size, 1)` instead of all `input_ids`
+            of shape `(batch_size, sequence_length)`.
+        inputs_embeds (`torch.FloatTensor` of shape `(batch_size, sequence_length, hidden_size)`, *optional*):
+            Optionally, instead of passing `input_ids` you can choose to directly pass an embedded representation. This
+            is useful if you want more control over how to convert `input_ids` indices into associated vectors than the
+            model's internal embedding lookup matrix.
+        use_cache (`bool`, *optional*):
+            If set to `True`, `past_key_values` key value states are returned and can be used to speed up decoding (see
+            `past_key_values`).
+        output_attentions (`bool`, *optional*):
+            Whether or not to return the attentions tensors of all attention layers. See `attentions` under returned
+            tensors for more detail.
+        output_hidden_states (`bool`, *optional*):
+            Whether or not to return the hidden states of all layers. See `hidden_states` under returned tensors for
+            more detail.
+        return_dict (`bool`, *optional*):
+            Whether or not to return a [`~utils.ModelOutput`] instead of a plain tuple.
+"""
+
+
+@add_start_docstrings(
+    "The bare LLaMA Model outputting raw hidden-states without any specific head on top.",
+    LLAMA_START_DOCSTRING,
+)
+class LlamaModel(LlamaPreTrainedModel):
+    """
+    Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`LlamaDecoderLayer`]
+
+    Args:
+        config: LlamaConfig
+    """
+
+    def __init__(self, config: LlamaConfig):
+        super().__init__(config)
+        self.padding_idx = config.pad_token_id
+        self.vocab_size = config.vocab_size
+
+        self.embed_tokens = nn.Embedding(config.vocab_size, config.hidden_size, self.padding_idx)
+        self.layers = nn.ModuleList(
+            [LlamaDecoderLayer(config, layer_idx) for layer_idx in range(config.num_hidden_layers)]
+        )
+        self.norm = LlamaRMSNorm(config.hidden_size, eps=config.rms_norm_eps)
+        self.gradient_checkpointing = False
+
+        # register a causal mask to separate causal and padding mask creation. Merging happends in the attention class
+        causal_mask = torch.full((config.max_position_embeddings, config.max_position_embeddings), fill_value=1)
+        self.register_buffer("causal_mask", torch.triu(causal_mask, diagonal=1), persistent=False)
+        # Initialize weights and apply final processing
+        self.post_init()
+
+    def get_input_embeddings(self):
+        return self.embed_tokens
+
+    def set_input_embeddings(self, value):
+        self.embed_tokens = value
+
+    @add_start_docstrings_to_model_forward(LLAMA_INPUTS_DOCSTRING)
+    def forward(
+        self,
+        input_ids: torch.LongTensor = None,
+        attention_mask: Optional[torch.Tensor] = None,
+        position_ids: Optional[torch.LongTensor] = None,
+        past_key_values: Optional[List[torch.FloatTensor]] = None,
+        inputs_embeds: Optional[torch.FloatTensor] = None,
+        use_cache: Optional[bool] = None,
+        output_attentions: Optional[bool] = None,
+        output_hidden_states: Optional[bool] = None,
+        return_dict: Optional[bool] = None,
+        cache_position: Optional[torch.LongTensor] = None,
+    ) -> Union[Tuple, BaseModelOutputWithPast]:
+        output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
+        output_hidden_states = (
+            output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
+        )
+        use_cache = use_cache if use_cache is not None else self.config.use_cache
+        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+        # use_cache = False
+        if (input_ids is None) ^ (inputs_embeds is not None):
+            raise ValueError(
+                "You cannot specify both input_ids and inputs_embeds at the same time, and must specify either one"
+            )
+
+        if self.gradient_checkpointing and self.training and use_cache:
+            logger.warning_once(
+                "`use_cache=True` is incompatible with gradient checkpointing. Setting `use_cache=False`."
+            )
+            use_cache = False
+
+        if inputs_embeds is None:
+            inputs_embeds = self.embed_tokens(input_ids)
+
+        past_seen_tokens = 0
+        if use_cache:  # kept for BC (cache positions)
+            if not isinstance(past_key_values, StaticCache):
+                past_key_values = DynamicCache.from_legacy_cache(past_key_values)
+            past_seen_tokens = past_key_values.get_seq_length()
+
+        if cache_position is None:
+            cache_position = torch.arange(
+                past_seen_tokens, past_seen_tokens + inputs_embeds.shape[1], device=inputs_embeds.device
+            )
+
+        if position_ids is None:
+            position_ids = cache_position.unsqueeze(0)
+
+        causal_mask = self._update_causal_mask(attention_mask, inputs_embeds)
+
+        # embed positions
+        hidden_states = inputs_embeds
+
+        # decoder layers
+        all_hidden_states = () if output_hidden_states else None
+        all_self_attns = () if output_attentions else None
+        next_decoder_cache = None
+
+        for decoder_layer in self.layers:
+            if output_hidden_states:
+                all_hidden_states += (hidden_states,)
+
+            if self.gradient_checkpointing and self.training:
+                layer_outputs = self._gradient_checkpointing_func(
+                    decoder_layer.__call__,
+                    hidden_states,
+                    causal_mask,
+                    position_ids,
+                    past_key_values,
+                    output_attentions,
+                    use_cache,
+                    cache_position,
+                )
+            else:
+                layer_outputs = decoder_layer(
+                    hidden_states,
+                    attention_mask=causal_mask,
+                    position_ids=position_ids,
+                    past_key_value=past_key_values,
+                    output_attentions=output_attentions,
+                    use_cache=use_cache,
+                    cache_position=cache_position,
+                )
+
+            hidden_states = layer_outputs[0]
+
+            if use_cache and not decoder_layer.drop_attn:
+                next_decoder_cache = layer_outputs[2 if output_attentions else 1]
+
+            if output_attentions and not decoder_layer.drop_attn:
+                all_self_attns += (layer_outputs[1],)
+
+        hidden_states = self.norm(hidden_states)
+
+        # add hidden states from the last decoder layer
+        if output_hidden_states:
+            all_hidden_states += (hidden_states,)
+
+        next_cache = None
+        if use_cache:
+            next_cache = (
+                next_decoder_cache.to_legacy_cache() if isinstance(next_decoder_cache, Cache) else next_decoder_cache
+            )
+        # print(next_cache)
+        if not return_dict:
+            return tuple(v for v in [hidden_states, next_cache, all_hidden_states, all_self_attns] if v is not None)
+        return BaseModelOutputWithPast(
+            last_hidden_state=hidden_states,
+            past_key_values=next_cache,
+            hidden_states=all_hidden_states,
+            attentions=all_self_attns,
+        )
+
+    def _update_causal_mask(self, attention_mask, input_tensor):
+        if self.config._attn_implementation == "flash_attention_2":
+            if attention_mask is not None and 0.0 in attention_mask:
+                return attention_mask
+            return None
+
+        batch_size, seq_length = input_tensor.shape[:2]
+        dtype = input_tensor.dtype
+        device = input_tensor.device
+
+        # support going beyond cached `max_position_embedding`
+        if seq_length > self.causal_mask.shape[-1]:
+            causal_mask = torch.full((2 * self.causal_mask.shape[-1], 2 * self.causal_mask.shape[-1]), fill_value=1)
+            self.register_buffer("causal_mask", torch.triu(causal_mask, diagonal=1), persistent=False)
+
+        if hasattr(self, "causal_mask"):  # we use the current dtype to avoid any overflows
+            causal_mask = (
+                self.causal_mask[None, None, :, :].repeat(batch_size, 1, 1, 1).to(dtype) * torch.finfo(dtype).min
+            )
+        else:
+            mask = torch.full(
+                (self.config.max_position_embeddings, self.config.max_position_embeddings),
+                fill_value=torch.finfo(dtype).min,
+            )
+            causal_mask = torch.triu(mask, diagonal=1)
+
+        causal_mask = causal_mask.to(dtype=dtype, device=device)
+        if attention_mask is not None and attention_mask.dim() == 2:
+            mask_length = attention_mask.shape[-1]
+            padding_mask = causal_mask[..., :mask_length].eq(0.0) * attention_mask[:, None, None, :].eq(0.0)
+            causal_mask[..., :mask_length] = causal_mask[..., :mask_length].masked_fill(
+                padding_mask, torch.finfo(dtype).min
+            )
+
+        if self.config._attn_implementation == "sdpa":
+            is_tracing = torch.jit.is_tracing() or isinstance(input_tensor, torch.fx.Proxy)
+            if not is_tracing and attention_mask is not None and torch.any(attention_mask != 1):
+                causal_mask = causal_mask.mul(~torch.all(causal_mask == causal_mask.min(), dim=-1)[..., None]).to(
+                    dtype
+                )
+
+        return causal_mask
+
+
+class LlamaForCausalLM(LlamaPreTrainedModel):
+    _tied_weights_keys = ["lm_head.weight"]
+
+    def __init__(self, config):
+        super().__init__(config)
+        self.model = LlamaModel(config)
+        self.vocab_size = config.vocab_size
+        self.lm_head = nn.Linear(config.hidden_size, config.vocab_size, bias=False)
+
+        # Initialize weights and apply final processing
+        self.post_init()
+
+    def get_input_embeddings(self):
+        return self.model.embed_tokens
+
+    def set_input_embeddings(self, value):
+        self.model.embed_tokens = value
+
+    def get_output_embeddings(self):
+        return self.lm_head
+
+    def set_output_embeddings(self, new_embeddings):
+        self.lm_head = new_embeddings
+
+    def set_decoder(self, decoder):
+        self.model = decoder
+
+    def get_decoder(self):
+        return self.model
+
+    @add_start_docstrings_to_model_forward(LLAMA_INPUTS_DOCSTRING)
+    @replace_return_docstrings(output_type=CausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
+    def forward(
+        self,
+        input_ids: torch.LongTensor = None,
+        attention_mask: Optional[torch.Tensor] = None,
+        position_ids: Optional[torch.LongTensor] = None,
+        past_key_values: Optional[List[torch.FloatTensor]] = None,
+        inputs_embeds: Optional[torch.FloatTensor] = None,
+        labels: Optional[torch.LongTensor] = None,
+        use_cache: Optional[bool] = None,
+        output_attentions: Optional[bool] = None,
+        output_hidden_states: Optional[bool] = None,
+        return_dict: Optional[bool] = None,
+        cache_position: Optional[torch.LongTensor] = None,
+    ) -> Union[Tuple, CausalLMOutputWithPast]:
+        r"""
+        Args:
+            labels (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*):
+                Labels for computing the masked language modeling loss. Indices should either be in `[0, ...,
+                config.vocab_size]` or -100 (see `input_ids` docstring). Tokens with indices set to `-100` are ignored
+                (masked), the loss is only computed for the tokens with labels in `[0, ..., config.vocab_size]`.
+
+        Returns:
+
+        Example:
+
+        ```python
+        >>> from transformers import AutoTokenizer, LlamaForCausalLM
+
+        >>> model = LlamaForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf")
+        >>> tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
+
+        >>> prompt = "Hey, are you conscious? Can you talk to me?"
+        >>> inputs = tokenizer(prompt, return_tensors="pt")
+
+        >>> # Generate
+        >>> generate_ids = model.generate(inputs.input_ids, max_length=30)
+        >>> tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
+        "Hey, are you conscious? Can you talk to me?\nI'm not conscious, but I can talk to you."
+        ```"""
+        output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
+        output_hidden_states = (
+            output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
+        )
+        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+
+        # decoder outputs consists of (dec_features, layer_state, dec_hidden, dec_attn)
+        outputs = self.model(
+            input_ids=input_ids,
+            attention_mask=attention_mask,
+            position_ids=position_ids,
+            past_key_values=past_key_values,
+            inputs_embeds=inputs_embeds,
+            use_cache=use_cache,
+            output_attentions=output_attentions,
+            output_hidden_states=output_hidden_states,
+            return_dict=return_dict,
+            cache_position=cache_position,
+        )
+
+        hidden_states = outputs[0]
+        if self.config.pretraining_tp > 1:
+            lm_head_slices = self.lm_head.weight.split(self.vocab_size // self.config.pretraining_tp, dim=0)
+            logits = [F.linear(hidden_states, lm_head_slices[i]) for i in range(self.config.pretraining_tp)]
+            logits = torch.cat(logits, dim=-1)
+        else:
+            logits = self.lm_head(hidden_states)
+        logits = logits.float()
+
+        loss = None
+        if labels is not None:
+            # Shift so that tokens < n predict n
+            shift_logits = logits[..., :-1, :].contiguous()
+            shift_labels = labels[..., 1:].contiguous()
+            # Flatten the tokens
+            loss_fct = CrossEntropyLoss()
+            shift_logits = shift_logits.view(-1, self.config.vocab_size)
+            shift_labels = shift_labels.view(-1)
+            # Enable model parallelism
+            shift_labels = shift_labels.to(shift_logits.device)
+            loss = loss_fct(shift_logits, shift_labels)
+
+        if not return_dict:
+            output = (logits,) + outputs[1:]
+            return (loss,) + output if loss is not None else output
+
+        return CausalLMOutputWithPast(
+            loss=loss,
+            logits=logits,
+            past_key_values=outputs.past_key_values,
+            hidden_states=outputs.hidden_states,
+            attentions=outputs.attentions,
+        )
+
+    def prepare_inputs_for_generation(
+        self, input_ids, past_key_values=None, attention_mask=None, inputs_embeds=None, **kwargs
+    ):
+        past_length = 0
+        if past_key_values is not None:
+            if isinstance(past_key_values, Cache):
+                cache_length = past_key_values.get_seq_length()
+                past_length = past_key_values.seen_tokens
+                max_cache_length = past_key_values.get_max_length()
+            else:
+                cache_length = past_length = past_key_values[0][0].shape[2]
+                max_cache_length = None
+
+            # Keep only the unprocessed tokens:
+            # 1 - If the length of the attention_mask exceeds the length of input_ids, then we are in a setting where
+            # some of the inputs are exclusively passed as part of the cache (e.g. when passing input_embeds as
+            # input)
+            if attention_mask is not None and attention_mask.shape[1] > input_ids.shape[1]:
+                input_ids = input_ids[:, -(attention_mask.shape[1] - past_length) :]
+            # 2 - If the past_length is smaller than input_ids', then input_ids holds all input tokens. We can discard
+            # input_ids based on the past_length.
+            elif past_length < input_ids.shape[1]:
+                input_ids = input_ids[:, past_length:]
+            # 3 - Otherwise (past_length >= input_ids.shape[1]), let's assume input_ids only has unprocessed tokens.
+
+            # If we are about to go beyond the maximum cache length, we need to crop the input attention mask.
+            if (
+                max_cache_length is not None
+                and attention_mask is not None
+                and cache_length + input_ids.shape[1] > max_cache_length
+            ):
+                attention_mask = attention_mask[:, -max_cache_length:]
+
+        position_ids = kwargs.get("position_ids", None)
+        if attention_mask is not None and position_ids is None:
+            # create position_ids on the fly for batch generation
+            position_ids = attention_mask.long().cumsum(-1) - 1
+            position_ids.masked_fill_(attention_mask == 0, 1)
+            if past_key_values:
+                position_ids = position_ids[:, -input_ids.shape[1] :]
+
+        if past_key_value := getattr(self.model.layers[0].self_attn, "past_key_value", None):
+            # generation with static cache
+            past_length = past_key_value.get_seq_length()
+            input_ids = input_ids[:, past_length:]
+            position_ids = position_ids[:, past_length:]
+
+        # TODO @gante we should only keep a `cache_position` in generate, and do +=1.
+        # same goes for position ids. Could also help with continued generation.
+        cache_position = kwargs.get("cache_position", None)
+        if cache_position is None:
+            cache_position = torch.arange(
+                past_length, past_length + position_ids.shape[-1], device=position_ids.device
+            )
+
+        # if `inputs_embeds` are passed, we only want to use them in the 1st generation step
+        if inputs_embeds is not None and past_key_values is None:
+            model_inputs = {"inputs_embeds": inputs_embeds}
+        else:
+            model_inputs = {"input_ids": input_ids}
+
+        model_inputs.update(
+            {
+                "position_ids": position_ids,
+                "cache_position": cache_position,
+                "past_key_values": past_key_values,
+                "use_cache": kwargs.get("use_cache"),
+                "attention_mask": attention_mask,
+            }
+        )
+        return model_inputs
+
+    @staticmethod
+    def _reorder_cache(past_key_values, beam_idx):
+        reordered_past = ()
+        for layer_past in past_key_values:
+            reordered_past += (
+                tuple(past_state.index_select(0, beam_idx.to(past_state.device)) for past_state in layer_past),
+            )
+        return reordered_past
+
+
+@add_start_docstrings(
+    """
+    The LLaMa Model transformer with a sequence classification head on top (linear layer).
+
+    [`LlamaForSequenceClassification`] uses the last token in order to do the classification, as other causal models
+    (e.g. GPT-2) do.
+
+    Since it does classification on the last token, it requires to know the position of the last token. If a
+    `pad_token_id` is defined in the configuration, it finds the last token that is not a padding token in each row. If
+    no `pad_token_id` is defined, it simply takes the last value in each row of the batch. Since it cannot guess the
+    padding tokens when `inputs_embeds` are passed instead of `input_ids`, it does the same (take the last value in
+    each row of the batch).
+    """,
+    LLAMA_START_DOCSTRING,
+)
+class LlamaForSequenceClassification(LlamaPreTrainedModel):
+    def __init__(self, config):
+        super().__init__(config)
+        self.num_labels = config.num_labels
+        self.model = LlamaModel(config)
+        self.score = nn.Linear(config.hidden_size, self.num_labels, bias=False)
+
+        # Initialize weights and apply final processing
+        self.post_init()
+
+    def get_input_embeddings(self):
+        return self.model.embed_tokens
+
+    def set_input_embeddings(self, value):
+        self.model.embed_tokens = value
+
+    @add_start_docstrings_to_model_forward(LLAMA_INPUTS_DOCSTRING)
+    def forward(
+        self,
+        input_ids: torch.LongTensor = None,
+        attention_mask: Optional[torch.Tensor] = None,
+        position_ids: Optional[torch.LongTensor] = None,
+        past_key_values: Optional[List[torch.FloatTensor]] = None,
+        inputs_embeds: Optional[torch.FloatTensor] = None,
+        labels: Optional[torch.LongTensor] = None,
+        use_cache: Optional[bool] = None,
+        output_attentions: Optional[bool] = None,
+        output_hidden_states: Optional[bool] = None,
+        return_dict: Optional[bool] = None,
+    ) -> Union[Tuple, SequenceClassifierOutputWithPast]:
+        r"""
+        labels (`torch.LongTensor` of shape `(batch_size,)`, *optional*):
+            Labels for computing the sequence classification/regression loss. Indices should be in `[0, ...,
+            config.num_labels - 1]`. If `config.num_labels == 1` a regression loss is computed (Mean-Square loss), If
+            `config.num_labels > 1` a classification loss is computed (Cross-Entropy).
+        """
+        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+
+        transformer_outputs = self.model(
+            input_ids,
+            attention_mask=attention_mask,
+            position_ids=position_ids,
+            past_key_values=past_key_values,
+            inputs_embeds=inputs_embeds,
+            use_cache=use_cache,
+            output_attentions=output_attentions,
+            output_hidden_states=output_hidden_states,
+            return_dict=return_dict,
+        )
+        hidden_states = transformer_outputs[0]
+        logits = self.score(hidden_states)
+
+        if input_ids is not None:
+            batch_size = input_ids.shape[0]
+        else:
+            batch_size = inputs_embeds.shape[0]
+
+        if self.config.pad_token_id is None and batch_size != 1:
+            raise ValueError("Cannot handle batch sizes > 1 if no padding token is defined.")
+        if self.config.pad_token_id is None:
+            sequence_lengths = -1
+        else:
+            if input_ids is not None:
+                # if no pad token found, use modulo instead of reverse indexing for ONNX compatibility
+                sequence_lengths = torch.eq(input_ids, self.config.pad_token_id).int().argmax(-1) - 1
+                sequence_lengths = sequence_lengths % input_ids.shape[-1]
+                sequence_lengths = sequence_lengths.to(logits.device)
+            else:
+                sequence_lengths = -1
+
+        pooled_logits = logits[torch.arange(batch_size, device=logits.device), sequence_lengths]
+
+        loss = None
+        if labels is not None:
+            labels = labels.to(logits.device)
+            if self.config.problem_type is None:
+                if self.num_labels == 1:
+                    self.config.problem_type = "regression"
+                elif self.num_labels > 1 and (labels.dtype == torch.long or labels.dtype == torch.int):
+                    self.config.problem_type = "single_label_classification"
+                else:
+                    self.config.problem_type = "multi_label_classification"
+
+            if self.config.problem_type == "regression":
+                loss_fct = MSELoss()
+                if self.num_labels == 1:
+                    loss = loss_fct(pooled_logits.squeeze(), labels.squeeze())
+                else:
+                    loss = loss_fct(pooled_logits, labels)
+            elif self.config.problem_type == "single_label_classification":
+                loss_fct = CrossEntropyLoss()
+                loss = loss_fct(pooled_logits.view(-1, self.num_labels), labels.view(-1))
+            elif self.config.problem_type == "multi_label_classification":
+                loss_fct = BCEWithLogitsLoss()
+                loss = loss_fct(pooled_logits, labels)
+        if not return_dict:
+            output = (pooled_logits,) + transformer_outputs[1:]
+            return ((loss,) + output) if loss is not None else output
+
+        return SequenceClassifierOutputWithPast(
+            loss=loss,
+            logits=pooled_logits,
+            past_key_values=transformer_outputs.past_key_values,
+            hidden_states=transformer_outputs.hidden_states,
+            attentions=transformer_outputs.attentions,
+        )
+
+
+@add_start_docstrings(
+    """
+The Llama Model transformer with a span classification head on top for extractive question-answering tasks like
+SQuAD (a linear layer on top of the hidden-states output to compute `span start logits` and `span end logits`).
+    """,
+    LLAMA_START_DOCSTRING,
+)
+class LlamaForQuestionAnswering(LlamaPreTrainedModel):
+    # Copied from transformers.models.bloom.modeling_bloom.BloomForQuestionAnswering.__init__ with Bloom->Llama
+    def __init__(self, config):
+        super().__init__(config)
+        self.transformer = LlamaModel(config)
+        self.qa_outputs = nn.Linear(config.hidden_size, 2)
+
+        # Initialize weights and apply final processing
+        self.post_init()
+
+    def get_input_embeddings(self):
+        return self.transformer.embed_tokens
+
+    def set_input_embeddings(self, value):
+        self.transformer.embed_tokens = value
+
+    @add_start_docstrings_to_model_forward(LLAMA_INPUTS_DOCSTRING)
+    def forward(
+        self,
+        input_ids: Optional[torch.LongTensor] = None,
+        attention_mask: Optional[torch.FloatTensor] = None,
+        position_ids: Optional[torch.LongTensor] = None,
+        past_key_values: Optional[List[torch.FloatTensor]] = None,
+        inputs_embeds: Optional[torch.FloatTensor] = None,
+        start_positions: Optional[torch.LongTensor] = None,
+        end_positions: Optional[torch.LongTensor] = None,
+        output_attentions: Optional[bool] = None,
+        output_hidden_states: Optional[bool] = None,
+        return_dict: Optional[bool] = None,
+    ) -> Union[Tuple, QuestionAnsweringModelOutput]:
+        r"""
+        start_positions (`torch.LongTensor` of shape `(batch_size,)`, *optional*):
+            Labels for position (index) of the start of the labelled span for computing the token classification loss.
+            Positions are clamped to the length of the sequence (`sequence_length`). Position outside of the sequence
+            are not taken into account for computing the loss.
+        end_positions (`torch.LongTensor` of shape `(batch_size,)`, *optional*):
+            Labels for position (index) of the end of the labelled span for computing the token classification loss.
+            Positions are clamped to the length of the sequence (`sequence_length`). Position outside of the sequence
+            are not taken into account for computing the loss.
+        """
+        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+
+        outputs = self.transformer(
+            input_ids,
+            attention_mask=attention_mask,
+            position_ids=position_ids,
+            past_key_values=past_key_values,
+            inputs_embeds=inputs_embeds,
+            output_attentions=output_attentions,
+            output_hidden_states=output_hidden_states,
+            return_dict=return_dict,
+        )
+
+        sequence_output = outputs[0]
+
+        logits = self.qa_outputs(sequence_output)
+        start_logits, end_logits = logits.split(1, dim=-1)
+        start_logits = start_logits.squeeze(-1).contiguous()
+        end_logits = end_logits.squeeze(-1).contiguous()
+
+        total_loss = None
+        if start_positions is not None and end_positions is not None:
+            # If we are on multi-GPU, split add a dimension
+            if len(start_positions.size()) > 1:
+                start_positions = start_positions.squeeze(-1).to(start_logits.device)
+            if len(end_positions.size()) > 1:
+                end_positions = end_positions.squeeze(-1).to(end_logits.device)
+            # sometimes the start/end positions are outside our model inputs, we ignore these terms
+            ignored_index = start_logits.size(1)
+            start_positions = start_positions.clamp(0, ignored_index)
+            end_positions = end_positions.clamp(0, ignored_index)
+
+            loss_fct = CrossEntropyLoss(ignore_index=ignored_index)
+            start_loss = loss_fct(start_logits, start_positions)
+            end_loss = loss_fct(end_logits, end_positions)
+            total_loss = (start_loss + end_loss) / 2
+
+        if not return_dict:
+            output = (start_logits, end_logits) + outputs[2:]
+            return ((total_loss,) + output) if total_loss is not None else output
+
+        return QuestionAnsweringModelOutput(
+            loss=total_loss,
+            start_logits=start_logits,
+            end_logits=end_logits,
+            hidden_states=outputs.hidden_states,
+            attentions=outputs.attentions,
+        )

special_tokens_map.json

@@ -0,0 +1,23 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer.model

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

tokenizer_config.json

@@ -0,0 +1,35 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "bos_token": {
+    "__type": "AddedToken",
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "clean_up_tokenization_spaces": false,
+  "eos_token": {
+    "__type": "AddedToken",
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "legacy": false,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": null,
+  "padding_side": "right",
+  "sp_model_kwargs": {},
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": {
+    "__type": "AddedToken",
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}