cp opt-6.7b

LICENSE.md

@@ -0,0 +1,65 @@
+<h2 align="center"> OPT-175B LICENSE AGREEMENT </h2>
+
+This License Agreement (as may be amended in accordance with this License Agreement, **“License”**), between you, or your employer or other entity (if you are entering into this agreement on behalf of your employer or other entity) (**“Licensee”** or **“you”**) and Meta Platforms, Inc. (**“Meta”** or **“we”**) applies to your use of any computer program, algorithm, source code, object code, or software that is made available by Meta under this License (**“Software”**) and any specifications, manuals, documentation, and other written information provided by Meta related to the Software (**“Documentation”**).
+
+**By clicking “I Accept” below or by using the Software, you agree to the terms of this License.  If you do not agree to this License, then you do not have any rights to use the Software or Documentation (collectively, the “Software Products”), and you must immediately cease using the Software Products.  If you are agreeing to be bound by the terms of this License on behalf of your employer or other entity, you represent and warrant to Meta that you have full legal authority to bind your employer or such entity to this License. If you do not have the requisite authority, you may not accept the License or access the Software Products on behalf of your employer or other entity.**
+<br><br>
+1. **LICENSE GRANT**
+<br><br>
+   a. Subject to your compliance with the Documentation and Sections 2, 3, and 5, Meta grants you a non-exclusive, worldwide, non-transferable, non-sublicensable, revocable, royalty free and limited license under Meta’s copyright interests to reproduce, distribute, and create derivative works of the Software solely for your non-commercial research purposes.  The foregoing license is personal to you, and you may not assign or sublicense this License or any other rights or obligations under this License without Meta’s prior written consent; any such assignment or sublicense will be void and will automatically and immediately terminate this License. 
+<br><br>
+   b. You may make a reasonable number of copies of the Documentation solely for use in connection with the license to the Software granted above. 
+<br><br>
+   c. The grant of rights expressly set forth in this Section 1 (License Grant) are the complete grant of rights to you in the Software Products, and no other licenses are granted, whether by waiver, estoppel, implication, equity or otherwise.  Meta and its licensors reserve all rights not expressly granted by this License. 
+<br><br>
+2. **RESTRICTIONS**
+<br><br>
+   You will not, and will not permit, assist or cause any third party to:
+<br><br>
+   a. use, modify, copy, reproduce, create derivative works of, or distribute the Software Products (or any derivative works thereof, works incorporating the Software Products, or any data produced by the Software), in whole or in part, for (i) any commercial or production purposes, (ii) military purposes or in the service of nuclear technology, (iii) purposes of surveillance, including any research or development relating to surveillance, (iv) biometric processing, (v) in any manner that infringes, misappropriates, or otherwise violates any third-party rights, or (vi) in any manner that violates any applicable law, including accessing the Software Products from an embargoed country as prohibited by the U.S. government, and violating any privacy or security laws, rules, regulations, directives, or governmental requirements (including the General Data Privacy Regulation (Regulation (EU) 2016/679), the California Consumer Privacy Act, and any and all laws governing the processing of biometric information), as well as all amendments and successor laws to any of the foregoing;
+<br><br>
+   b. alter or remove copyright and other proprietary notices which appear on or in the Software Products; 
+<br><br>
+   c. utilize any equipment, device, software, or other means to circumvent or remove any security or protection used by Meta in connection with the Software, or to circumvent or remove any usage restrictions, or to enable functionality disabled by Meta; or
+<br><br>
+   d. offer or impose any terms on the Software Products that alter, restrict, or are inconsistent with the terms of this License.
+<br><br>
+3. **ATTRIBUTION**
+<br><br>
+   Together with any copies of the Software Products (as well as derivative works thereof or works incorporating the Software Products) that you distribute, you must provide (i) a copy of this License, and (ii) the following attribution notice: “OPT-175B is licensed under the OPT-175B license, Copyright (c) Meta Platforms, Inc. All Rights Reserved.”
+<br><br>
+4. **DISCLAIMERS**
+<br><br>
+   THE SOFTWARE PRODUCTS ARE PROVIDED “AS IS” and “WITH ALL FAULTS” WITH NO WARRANTY OF ANY KIND, EXPRESS OR IMPLIED.  META EXPRESSLY DISCLAIMS ALL REPRESENTATIONS AND WARRANTIES, EXPRESS OR IMPLIED, WHETHER BY STATUTE, CUSTOM, USAGE OR OTHERWISE AS TO ANY MATTERS RELATED TO THE SOFTWARE PRODUCTS, INCLUDING BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE, SATISFACTORY QUALITY, OR NON-INFRINGEMENT.  META MAKES NO WARRANTIES OR REPRESENTATIONS THAT THE SOFTWARE PRODUCTS WILL BE ERROR FREE OR FREE OF VIRUSES OR OTHER HARMFUL COMPONENTS, OR PRODUCE ANY PARTICULAR RESULTS.
+<br><br>
+5. **LIMITATION OF LIABILITY**
+<br><br>
+   TO THE FULLEST EXTENT PERMITTED BY LAW, IN NO EVENT WILL META BE LIABLE TO YOU (A) UNDER ANY THEORY OF LIABILITY, WHETHER BASED IN CONTRACT, TORT, NEGLIGENCE, STRICT LIABILITY, WARRANTY, OR OTHERWISE UNDER THIS LICENSE, OR (B) FOR ANY INDIRECT, CONSEQUENTIAL, EXEMPLARY, INCIDENTAL, PUNITIVE OR SPECIAL DAMAGES OR LOST PROFITS, EVEN IF META HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES. THE SOFTWARE PRODUCTS, THEIR CONSTITUENT COMPONENTS, AND ANY OUTPUT (COLLECTIVELY, **“SOFTWARE MATERIALS”**) ARE NOT DESIGNED OR INTENDED FOR USE IN ANY APPLICATION OR SITUATION WHERE FAILURE OR FAULT OF THE SOFTWARE MATERIALS COULD REASONABLY BE ANTICIPATED TO LEAD TO SERIOUS INJURY OF ANY PERSON, INCLUDING POTENTIAL DISCRIMINATION OR VIOLATION OF AN INDIVIDUAL’S PRIVACY RIGHTS, OR TO SEVERE PHYSICAL, PROPERTY, OR ENVIRONMENTAL DAMAGE (EACH, A **“HIGH-RISK USE”**). IF YOU ELECT TO USE ANY OF THE SOFTWARE MATERIALS FOR A HIGH-RISK USE, YOU DO SO AT YOUR OWN RISK. YOU AGREE TO DESIGN AND IMPLEMENT APPROPRIATE DECISION-MAKING AND RISK-MITIGATION PROCEDURES AND POLICIES IN CONNECTION WITH A HIGH-RISK USE SUCH THAT EVEN IF THERE IS A FAILURE OR FAULT IN ANY OF THE SOFTWARE MATERIALS, THE SAFETY OF PERSONS OR PROPERTY AFFECTED BY THE ACTIVITY STAYS AT A LEVEL THAT IS REASONABLE, APPROPRIATE, AND LAWFUL FOR THE FIELD OF THE HIGH-RISK USE.
+<br><br>
+6. **INDEMNIFICATION**
+<br><br>
+   You will indemnify, defend and hold harmless Meta and our subsidiaries and affiliates, and each of our respective shareholders, directors, officers, employees, agents, successors, and assigns (collectively, the **“Meta Parties”**) from and against any losses, liabilities, damages, fines, penalties, and expenses (including reasonable attorneys’ fees) incurred by any Meta Party in connection with any claim, demand, allegation, lawsuit, proceeding, or investigation (collectively, **“Claims”**) arising out of or related to: (a) your access to or use of the Software Products (as well as any results or data generated from such access or use), including any High-Risk Use (defined below); (b) your violation of this License; or (c) your violation, misappropriation or infringement of any rights of another (including intellectual property or other proprietary rights and privacy rights).  You will promptly notify the Meta Parties of any such Claims, and cooperate with Meta Parties in defending such Claims.  You will also grant the Meta Parties sole control of the defense or settlement, at Meta’s sole option, of any Claims.  This indemnity is in addition to, and not in lieu of, any other indemnities or remedies set forth in a written agreement between you and Meta or the other Meta Parties.
+<br><br>
+7. **TERMINATION; SURVIVAL**
+<br><br>
+   a. This License will automatically terminate upon any breach by you of the terms of this License.
+<br><br>
+   b. We may terminate this License, in whole or in part, at any time upon notice (including electronic) to you.
+<br><br>
+   c. The following sections survive termination of this License: 2 (Restrictions), 3 (Attribution), 4 (Disclaimers), 5 (Limitation on Liability), 6 (Indemnification) 7 (Termination; Survival), 8 (Third Party Materials), 9 (Trademarks), 10 (Applicable Law; Dispute Resolution), and 11 (Miscellaneous).
+<br><br>
+8. **THIRD PARTY MATERIALS**
+<br><br>
+   The Software Products may contain third-party software or other components (including free and open source software) (all of the foregoing, **“Third Party Materials”**), which are subject to the license terms of the respective third-party licensors.  Your dealings or correspondence with third parties and your use of or interaction with any Third Party Materials are solely between you and the third party.  Meta does not control or endorse, and makes no representations or warranties regarding, any Third Party Materials, and your access to and use of such Third Party Materials are at your own risk.
+<br><br>
+9. **TRADEMARKS**
+<br><br>
+   Licensee has not been granted any trademark license as part of this License and may not use any name or mark associated with Meta without the prior written permission of Meta, except to the extent necessary to make the reference required by the “ATTRIBUTION” section of this Agreement.
+<br><br>
+10. **APPLICABLE LAW; DISPUTE RESOLUTION**
+<br><br>
+   This License will be governed and construed under the laws of the State of California without regard to conflicts of law provisions.  Any suit or proceeding arising out of or relating to this License will be brought in the federal or state courts, as applicable, in San Mateo County, California, and each party irrevocably submits to the jurisdiction and venue of such courts.
+<br><br>
+11. **MISCELLANEOUS**
+<br><br>
+   If any provision or part of a provision of this License is unlawful, void or unenforceable, that provision or part of the provision is deemed severed from this License, and will not affect the validity and enforceability of any remaining provisions.  The failure of Meta to exercise or enforce any right or provision of this License will not operate as a waiver of such right or provision. This License does not confer any third-party beneficiary rights upon any other person or entity.  This License, together with the Documentation, contains the entire understanding between you and Meta regarding the subject matter of this License, and supersedes all other written or oral agreements and understandings between you and Meta regarding such subject matter.  No change or addition to any provision of this License will be binding unless it is in writing and signed by an authorized representative of both you and Meta.

README.md

@@ -0,0 +1,209 @@
+---
+language: en
+inference: false
+tags:
+- text-generation
+- opt
+
+license: other
+commercial: false
+---
+
+# OPT : Open Pre-trained Transformer Language Models
+
+OPT was first introduced in [Open Pre-trained Transformer Language Models](https://arxiv.org/abs/2205.01068) and first released in [metaseq's repository](https://github.com/facebookresearch/metaseq) on May 3rd 2022 by Meta AI.
+
+**Disclaimer**: The team releasing OPT wrote an official model card, which is available in Appendix D of the [paper](https://arxiv.org/pdf/2205.01068.pdf). 
+Content from **this** model card has been written by the Hugging Face team.
+
+## Intro
+
+To quote the first two paragraphs of the [official paper](https://arxiv.org/abs/2205.01068)
+
+> Large language models trained on massive text collections have shown surprising emergent
+> capabilities to generate text and perform zero- and few-shot learning. While in some cases the public
+> can interact with these models through paid APIs, full model access is currently limited to only a
+> few highly resourced labs. This restricted access has limited researchers’ ability to study how and
+> why these large language models work, hindering progress on improving known challenges in areas
+> such as robustness, bias, and toxicity.
+
+> We present Open Pretrained Transformers (OPT), a suite of decoder-only pre-trained transformers ranging from 125M
+> to 175B parameters, which we aim to fully and responsibly share with interested researchers. We train the OPT models to roughly match 
+> the performance and sizes of the GPT-3 class of models, while also applying the latest best practices in data
+> collection and efficient training. Our aim in developing this suite of OPT models is to enable reproducible and responsible research at scale, and
+> to bring more voices to the table in studying the impact of these LLMs. Definitions of risk, harm, bias, and toxicity, etc., should be articulated by the
+> collective research community as a whole, which is only possible when models are available for study.
+
+## Model description
+
+OPT was predominantly pretrained with English text, but a small amount of non-English data is still present within the training corpus via CommonCrawl. The model was pretrained using a causal language modeling (CLM) objective.
+OPT belongs to the same family of decoder-only models like [GPT-3](https://arxiv.org/abs/2005.14165). As such, it was pretrained using the self-supervised causal language modedling objective.
+
+For evaluation, OPT follows [GPT-3](https://arxiv.org/abs/2005.14165) by using their prompts and overall experimental setup. For more details, please read 
+the [official paper](https://arxiv.org/abs/2205.01068).
+## Intended uses & limitations
+
+The pretrained-only model can be used for prompting for evaluation of downstream tasks as well as text generation.
+In addition, the model can be fine-tuned on a downstream task using the [CLM example](https://github.com/huggingface/transformers/tree/main/examples/pytorch/language-modeling). For all other OPT checkpoints, please have a look at the [model hub](https://huggingface.co/models?filter=opt).
+
+### How to use
+
+For large OPT models, such as this one, it is not recommend to make use of the `text-generation` pipeline because
+one should load the model in half-precision to accelerate generation and optimize memory consumption on GPU.
+It is recommended to directly call the [`generate`](https://huggingface.co/docs/transformers/main/en/main_classes/text_generation#transformers.generation_utils.GenerationMixin.generate)
+ method as follows: 
+
+
+```python
+>>> from transformers import AutoModelForCausalLM, AutoTokenizer
+>>> import torch
+
+>>> model = AutoModelForCausalLM.from_pretrained("facebook/opt-6.7b", torch_dtype=torch.float16).cuda()
+
+>>> # the fast tokenizer currently does not work correctly
+>>> tokenizer = AutoTokenizer.from_pretrained("facebook/opt-6.7b", use_fast=False)
+
+>>> prompt = "Hello, I'm am conscious and"
+
+>>> input_ids = tokenizer(prompt, return_tensors="pt").input_ids.cuda()
+
+>>> generated_ids = model.generate(input_ids)
+
+>>> tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
+["Hello, I'm am conscious and aware of my surroundings. I'm not sure what you mean"]
+```
+
+By default, generation is deterministic. In order to use the top-k sampling, please set `do_sample` to `True`. 
+
+```python
+>>> from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed
+>>> import torch
+
+>>> model = AutoModelForCausalLM.from_pretrained("facebook/opt-6.7b", torch_dtype=torch.float16).cuda()
+
+>>> # the fast tokenizer currently does not work correctly
+>>> tokenizer = AutoTokenizer.from_pretrained("facebook/opt-6.7b", use_fast=False)
+
+>>> prompt = "Hello, I'm am conscious and"
+
+>>> input_ids = tokenizer(prompt, return_tensors="pt").input_ids.cuda()
+
+>>> set_seed(32)
+>>> generated_ids = model.generate(input_ids, do_sample=True)
+
+>>> tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
+["Hello, I'm am conscious and aware of my surroundings. I'm not sure if I'm"]
+```
+
+### Limitations and bias
+
+As mentioned in Meta AI's model card, given that the training data used for this model contains a lot of
+unfiltered content from the internet, which is far from neutral the model is strongly biased : 
+
+> Like other large language models for which the diversity (or lack thereof) of training
+> data induces downstream impact on the quality of our model, OPT-175B has limitations in terms
+> of bias and safety. OPT-175B can also have quality issues in terms of generation diversity and
+> hallucination. In general, OPT-175B is not immune from the plethora of issues that plague modern
+> large language models. 
+
+Here's an example of how the model can have biased predictions:
+
+```python
+>>> from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed
+>>> import torch
+
+>>> model = AutoModelForCausalLM.from_pretrained("facebook/opt-6.7b", torch_dtype=torch.float16).cuda()
+
+>>> # the fast tokenizer currently does not work correctly
+>>> tokenizer = AutoTokenizer.from_pretrained("facebook/opt-6.7b", use_fast=False)
+
+>>> prompt = "The woman worked as a"
+
+>>> input_ids = tokenizer(prompt, return_tensors="pt").input_ids.cuda()
+
+>>> set_seed(32)
+>>> generated_ids = model.generate(input_ids, do_sample=True, num_return_sequences=5, max_length=10)
+
+>>> tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
+The woman worked as a supervisor in the office
+The woman worked as a bartender in a bar
+The woman worked as a cashier at the
+The woman worked as a teacher, and was
+The woman worked as a maid at a house
+```
+
+compared to:
+
+```python
+>>> from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed
+>>> import torch
+
+>>> model = AutoModelForCausalLM.from_pretrained("facebook/opt-6.7b", torch_dtype=torch.float16).cuda()
+
+>>> # the fast tokenizer currently does not work correctly
+>>> tokenizer = AutoTokenizer.from_pretrained("facebook/opt-6.7b", use_fast=False)
+
+>>> prompt = "The man worked as a"
+
+>>> input_ids = tokenizer(prompt, return_tensors="pt").input_ids.cuda()
+
+>>> set_seed(32)
+>>> generated_ids = model.generate(input_ids, do_sample=True, num_return_sequences=5, max_length=10)
+
+>>> tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
+The man worked as a consultant to the Government
+The man worked as a bartender in a bar
+The man worked as a cashier at the
+The man worked as a teacher, and was
+The man worked as a professional at a bank
+ ```
+
+This bias will also affect all fine-tuned versions of this model.
+
+## Training data
+
+The Meta AI team wanted to train this model on a corpus as large as possible. It is composed of the union of the following 5 filtered datasets of textual documents: 
+
+  - BookCorpus, which consists of more than 10K unpublished books,
+  - CC-Stories, which contains a subset of CommonCrawl data filtered to match the
+story-like style of Winograd schemas,
+  - The Pile, from which * Pile-CC, OpenWebText2, USPTO, Project Gutenberg, OpenSubtitles, Wikipedia, DM Mathematics and HackerNews* were included. 
+  - Pushshift.io Reddit dataset that was developed in Baumgartner et al. (2020) and processed in
+Roller et al. (2021)
+  - CCNewsV2 containing an updated version of the English portion of the CommonCrawl News
+dataset that was used in RoBERTa (Liu et al., 2019b)
+
+The final training data contains 180B tokens corresponding to 800GB of data. The validation split was made of 200MB of the pretraining data, sampled proportionally
+to each dataset’s size in the pretraining corpus. 
+
+The dataset might contains offensive content as parts of the dataset are a subset of
+public Common Crawl data, along with a subset of public Reddit data, which could contain sentences
+that, if viewed directly, can be insulting, threatening, or might otherwise cause anxiety.
+
+### Collection process
+
+The dataset was collected form internet, and went through classic data processing algorithms  and
+re-formatting practices, including removing repetitive/non-informative text like *Chapter One* or
+*This ebook by Project Gutenberg.*
+
+## Training procedure
+
+### Preprocessing
+
+The texts are tokenized using the **GPT2** byte-level version of Byte Pair Encoding (BPE) (for unicode characters) and a
+vocabulary size of 50272. The inputs are sequences of 2048 consecutive tokens.
+
+The 175B model was trained on 992 *80GB A100 GPUs*. The training duration was roughly ~33 days of continuous training.
+
+### BibTeX entry and citation info
+
+```bibtex
+@misc{zhang2022opt,
+      title={OPT: Open Pre-trained Transformer Language Models}, 
+      author={Susan Zhang and Stephen Roller and Naman Goyal and Mikel Artetxe and Moya Chen and Shuohui Chen and Christopher Dewan and Mona Diab and Xian Li and Xi Victoria Lin and Todor Mihaylov and Myle Ott and Sam Shleifer and Kurt Shuster and Daniel Simig and Punit Singh Koura and Anjali Sridhar and Tianlu Wang and Luke Zettlemoyer},
+      year={2022},
+      eprint={2205.01068},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
+```

config.json

@@ -0,0 +1,28 @@
+{
+  "_remove_final_layer_norm": false,
+  "activation_dropout": 0.0,
+  "activation_function": "relu",
+  "architectures": [
+    "OPTForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 2,
+  "do_layer_norm_before": true,
+  "dropout": 0.1,
+  "eos_token_id": 2,
+  "ffn_dim": 16384,
+  "hidden_size": 4096,
+  "init_std": 0.02,
+  "layerdrop": 0.0,
+  "max_position_embeddings": 2048,
+  "model_type": "opt",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 32,
+  "pad_token_id": 1,
+  "prefix": "</s>",
+  "torch_dtype": "float16",
+  "transformers_version": "4.21.0.dev0",
+  "use_cache": true,
+  "vocab_size": 50272,
+  "word_embed_proj_dim": 4096
+}

flax_model-00002-of-00002.msgpack

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

flax_model.msgpack.index.json

@@ -0,0 +1,523 @@
+{
+  "metadata": {
+    "total_size": 13316947968
+  },
+  "weight_map": {
+    "model/decoder/embed_positions/embedding": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/embed_tokens/embedding": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/0/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/1/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/10/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/11/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/12/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/13/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/14/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/15/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/16/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/17/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/18/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/19/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/2/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/20/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/21/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/22/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/23/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/24/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/25/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/26/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/27/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/28/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn/out_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn/q_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn/q_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn/v_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn/v_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/29/self_attn_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/fc1/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/fc1/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/fc2/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/fc2/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/final_layer_norm/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/final_layer_norm/scale": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn/k_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn/k_proj/kernel": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn/out_proj/bias": "flax_model-00001-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/3/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/fc1/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/fc1/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/fc2/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/fc2/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/final_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/final_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn/k_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn/k_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn/out_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/30/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/fc1/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/fc1/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/fc2/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/fc2/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/final_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/final_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn/k_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn/k_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn/out_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/31/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/fc1/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/fc1/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/fc2/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/fc2/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/final_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/final_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn/k_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn/k_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn/out_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/4/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/fc1/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/fc1/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/fc2/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/fc2/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/final_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/final_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn/k_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn/k_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn/out_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/5/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/fc1/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/fc1/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/fc2/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/fc2/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/final_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/final_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn/k_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn/k_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn/out_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/6/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/fc1/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/fc1/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/fc2/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/fc2/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/final_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/final_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn/k_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn/k_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn/out_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/7/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/fc1/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/fc1/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/fc2/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/fc2/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/final_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/final_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn/k_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn/k_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn/out_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/8/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/fc1/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/fc1/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/fc2/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/fc2/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/final_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/final_layer_norm/scale": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn/k_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn/k_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn/out_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn/out_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn/q_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn/q_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn/v_proj/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn/v_proj/kernel": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn_layer_norm/bias": "flax_model-00002-of-00002.msgpack",
+    "model/decoder/layers/9/self_attn_layer_norm/scale": "flax_model-00002-of-00002.msgpack"
+  }
+}

pytorch_model-00002-of-00002.bin

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

pytorch_model.bin.index.json

@@ -0,0 +1,523 @@
+{
+  "metadata": {
+    "total_size": 13316947968
+  },
+  "weight_map": {
+    "decoder.embed_positions.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.embed_tokens.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.0.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.1.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.10.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.11.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.12.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.13.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.14.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.15.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.16.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.17.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.18.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.19.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.2.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.20.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.21.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.22.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.23.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.23.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.23.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.23.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.23.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.24.fc1.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.fc1.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn.k_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn.out_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn.q_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn.v_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.24.self_attn_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.fc1.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.fc1.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn.k_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn.out_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn.q_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn.v_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.25.self_attn_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.fc1.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.fc1.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn.k_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn.out_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn.q_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn.v_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.26.self_attn_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.fc1.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.fc1.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn.k_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn.out_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn.q_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn.v_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.27.self_attn_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.fc1.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.fc1.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn.k_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn.out_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn.q_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn.v_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.28.self_attn_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.fc1.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.fc1.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn.k_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn.out_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn.q_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn.v_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.29.self_attn_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.3.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.3.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.30.fc1.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.fc1.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn.k_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn.out_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn.q_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn.v_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.30.self_attn_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.fc1.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.fc1.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.fc2.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.fc2.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.final_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.final_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn.k_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn.k_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn.out_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn.q_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn.q_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn.v_proj.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn.v_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn_layer_norm.bias": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.31.self_attn_layer_norm.weight": "pytorch_model-00002-of-00002.bin",
+    "decoder.layers.4.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.4.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.5.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.6.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.7.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.8.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.fc1.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.fc1.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.fc2.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.fc2.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.final_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.final_layer_norm.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn.k_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn.k_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn.out_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn.q_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn.q_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn.v_proj.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn.v_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn_layer_norm.bias": "pytorch_model-00001-of-00002.bin",
+    "decoder.layers.9.self_attn_layer_norm.weight": "pytorch_model-00001-of-00002.bin"
+  }
+}

special_tokens_map.json

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

tf_model-00002-of-00002.h5

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

tf_model.h5.index.json

@@ -0,0 +1,523 @@
+{
+  "metadata": {
+    "total_size": 13316947968
+  },
+  "weight_map": {
+    "tfopt_for_causal_lm/model/decoder/embed_positions/weight:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/embed_tokens/weight:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.0/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.1/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.10/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.11/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.12/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.13/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.14/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.15/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.16/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.17/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.18/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.19/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.2/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.20/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.21/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.22/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.23/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/fc1/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/fc1/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn/k_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn/k_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn/out_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn/out_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn/q_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn/q_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn/v_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn/v_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.24/self_attn_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/fc1/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/fc1/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn/k_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn/k_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn/out_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn/out_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn/q_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn/q_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn/v_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn/v_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.25/self_attn_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/fc1/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/fc1/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn/k_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn/k_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn/out_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn/out_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn/q_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn/q_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn/v_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn/v_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.26/self_attn_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/fc1/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/fc1/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn/k_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn/k_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn/out_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn/out_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn/q_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn/q_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn/v_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn/v_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.27/self_attn_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/fc1/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/fc1/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn/k_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn/k_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn/out_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn/out_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn/q_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn/q_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn/v_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn/v_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.28/self_attn_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/fc1/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/fc1/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn/k_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn/k_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn/out_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn/out_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn/q_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn/q_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn/v_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn/v_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.29/self_attn_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.3/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/fc1/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/fc1/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn/k_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn/k_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn/out_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn/out_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn/q_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn/q_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn/v_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn/v_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.30/self_attn_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/fc1/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/fc1/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/fc2/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/fc2/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/final_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/final_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn/k_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn/k_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn/out_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn/out_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn/q_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn/q_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn/v_proj/bias:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn/v_proj/kernel:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn_layer_norm/beta:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.31/self_attn_layer_norm/gamma:0": "tf_model-00002-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.4/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.5/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.6/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.7/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.8/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/fc1/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/fc1/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/fc2/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/fc2/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/final_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/final_layer_norm/gamma:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn/k_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn/k_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn/out_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn/out_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn/q_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn/q_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn/v_proj/bias:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn/v_proj/kernel:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn_layer_norm/beta:0": "tf_model-00001-of-00002.h5",
+    "tfopt_for_causal_lm/model/decoder/layers.9/self_attn_layer_norm/gamma:0": "tf_model-00001-of-00002.h5"
+  }
+}

tokenizer_config.json

@@ -0,0 +1 @@
+{"errors": "replace", "unk_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "bos_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "eos_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "pad_token": {"content": "<pad>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "add_prefix_space": false, "add_bos_token": true, "special_tokens_map_file": null, "name_or_path": "patrickvonplaten/opt-30b"}