初始化项目，由ModelHub XC社区提供模型

Model: lqtrung1998/galactica-6.7b-SFT-warmup-GSM8k
Source: Original Platform

.gitattributes

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

README.md

@@ -0,0 +1,76 @@
+---
+license: cc-by-nc-4.0
+---
+# ReFT: Reasoning with REinforced Fine-Tuning
+Paper: https://arxiv.org/pdf/2401.08967.pdf
+
+Repo: https://github.com/lqtrung1998/mwp_ReFT (under [Apache2.0 License](https://github.com/lqtrung1998/mwp_ReFT/blob/main/License.txt))
+
+## Introduction
+We introduce REinforced Fine-tuning (ReFT), a method that enhances the generalizability of learning LLMs for reasoning.
+
+This repository contains:
+- A Warmup Supervised Fine-tuned model on GSM8k benchmark: [lqtrung1998/galactica-6.7b-SFT-warmup-GSM8k](https://huggingface.co/lqtrung1998/galactica-6.7b-SFT-warmup-GSM8k)
+- A Supervised Fine-tuned model on GSM8k benchmark: [lqtrung1998/galactica-6.7b-SFT-GSM8k](https://huggingface.co/lqtrung1998/galactica-6.7b-SFT-GSM8k)
+- A Rerank model that can score the fine-tuned SFT model output: [lqtrung1998/galactica-6.7b-SFT-Rerank-GSM8k](https://huggingface.co/lqtrung1998/galactica-6.7b-SFT-Rerank-GSM8k)
+- A REinforced Fine-tuned model on GSM8k benchmark: [lqtrung1998/galactica-6.7b-ReFT-GSM8k](https://huggingface.co/lqtrung1998/galactica-6.7b-ReFT-GSM8k)
+- A Rerank model that can score the fine-tuned ReFT model output: [lqtrung1998/galactica-6.7b-ReFT-Rerank-GSM8k](https://huggingface.co/lqtrung1998/galactica-6.7b-ReFT-Rerank-GSM8k)
+
+Note: Our models are tuned based on Galactica, thus, licenses applicable to Galactica, such as non-commercial CC BY-NC 4.0 license also hold on these models.
+
+## Training Data
+The model is trained on GSM8k data with Python SDP CoT format, which can be found [here](https://github.com/lqtrung1998/mwp_ReFT)
+
+## Training Procedure
+Check out our paper and repo for complete details.
+#### ReFT model
+ReFT model is warm-up via Supervised Fine-tuning using GSM8k Python SDP training data for 2 epochs then it is REinforced Fine-tuned for 300 epochs using questions in GSM8k training set.
+#### Rerank model
+Rerank model is trained to classify if the output CoT is correct or not using sampling data of ReFT model after 2 epochs warm-up.
+
+## Evaluation Results
+See evaluations results of the models at table 4 of the research paper.
+
+Updated results:
+
+|                                                                    |  Top-1 | Voting@100 | Rerank@100 |
+|--------------------------------------------------------------------|:------:|:----------:|:----------:|
+| galactica-6.7b-SFT-warmup-GSM8k                                    |  48.37 |      -     |      -     |
+| galactica-6.7b-SFT-GSM8k<br>(+galactica-6.7b-SFT-Rerank-GSM8k)     | 58.83  |    62.9    |    73.4    |
+| galactica-6.7b-ReFT-GSM8k<br>(+galactica-6.7b-ReFT-Rerank-GSM8k)   |  68.91 |    71.9    |    76.4    |
+
+## Usage
+You can use the models through Huggingface's Transformers library or follow scripts in our repo.
+
+Prompt format:
+```python
+Question:
+Weng earns $12 an hour for babysitting. Yesterday, she
+just did 50 minutes of babysitting. How much did she earn?
+Answer reasoning:
+```
+Expected response:
+```python
+def solution():
+  """Weng earns $12 an hour for babysitting. Yesterday, she just did
+  50 minutes of babysitting. How much did she earn?"""
+  hourly_rate = 12
+  minutes_worked = 50
+  hours_worked = minutes_worked / 60
+  earnings = hourly_rate * hours_worked
+  result = earnings
+  return result
+```
+
+## Citation
+Please cite the paper if you use our data, model or code.
+```
+@misc{luong2024reft,
+      title={ReFT: Reasoning with Reinforced Fine-Tuning}, 
+      author={Trung Quoc Luong and Xinbo Zhang and Zhanming Jie and Peng Sun and Xiaoran Jin and Hang Li},
+      year={2024},
+      eprint={2401.08967},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
+```

config.json

@@ -0,0 +1,32 @@
+{
+  "_name_or_path": "/mnt/bn/trung-nas/hf_models/galactica-6.7b/",
+  "_remove_final_layer_norm": false,
+  "activation_dropout": 0.0,
+  "activation_function": "gelu",
+  "architectures": [
+    "OPTForCausalLM"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 0,
+  "do_layer_norm_before": true,
+  "dropout": 0.1,
+  "enable_bias": true,
+  "eos_token_id": 2,
+  "ffn_dim": 16384,
+  "hidden_size": 4096,
+  "init_std": 0.02,
+  "layer_norm_elementwise_affine": true,
+  "layerdrop": 0.0,
+  "learned_embeddings": true,
+  "max_position_embeddings": 2048,
+  "model_type": "opt",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 32,
+  "pad_token_id": 1,
+  "scale_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.34.1",
+  "use_cache": true,
+  "vocab_size": 50000,
+  "word_embed_proj_dim": 4096
+}

generation_config.json

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

pytorch_model-00001-of-00002.bin

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

pytorch_model-00002-of-00002.bin

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

pytorch_model.bin.index.json

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

special_tokens_map.json

@@ -0,0 +1,4 @@
+{
+  "eos_token": "</s>",
+  "pad_token": "<pad>"
+}

tokenizer_config.json

@@ -0,0 +1,193 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "4": {
+      "content": "[START_REF]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "5": {
+      "content": "[END_REF]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "6": {
+      "content": "[IMAGE]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "7": {
+      "content": "<fragments>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "8": {
+      "content": "</fragments>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "9": {
+      "content": "<work>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "10": {
+      "content": "</work>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "11": {
+      "content": "[START_SUP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "12": {
+      "content": "[END_SUP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "13": {
+      "content": "[START_SUB]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "14": {
+      "content": "[END_SUB]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "15": {
+      "content": "[START_DNA]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "16": {
+      "content": "[END_DNA]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "17": {
+      "content": "[START_AMINO]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "18": {
+      "content": "[END_AMINO]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "19": {
+      "content": "[START_SMILES]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "20": {
+      "content": "[END_SMILES]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "21": {
+      "content": "[START_I_SMILES]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "22": {
+      "content": "[END_I_SMILES]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "</s>",
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "tokenizer_class": "PreTrainedTokenizerFast"
+}