Initial model upload

.ipynb_checkpoints/README-checkpoint.md

@@ -0,0 +1,140 @@
+---
+library_name: peft
+license: apache-2.0
+base_model: ibm-granite/granite-3.1-8b-instruct
+tags:
+- generated_from_trainer
+model-index:
+- name: granite-bfcl-plans-3.1-8b-lora
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+[<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
+<details><summary>See axolotl config</summary>
+
+axolotl version: `0.5.2`
+```yaml
+base_model: ibm-granite/granite-3.1-8b-instruct
+model_type: AutoModelForCausalLM
+tokenizer_type: AutoTokenizer
+
+resize_token_embeddings_to_32x: true
+load_in_8bit: true
+load_in_4bit: false
+strict: false
+
+datasets:
+- path: bfcl_training_data.jsonl
+  type: chat_template
+  chat_template: tokenizer_default
+  field_messages: conversations
+  message_field_role: role
+  message_field_content: value
+dataset_prepared_path: last_run_prepared_sft_bfcl
+
+val_set_size: 0
+sequence_len: 16384
+sample_packing: false
+pad_to_sequence_len: true
+eval_sample_packing: false
+output_dir: granite-bfcl-plans-3.1-8b-lora
+
+wandb_project: null
+wandb_entity: null
+wandb_watch: null
+wandb_name: null
+wandb_log_model: null
+
+adapter: lora
+lora_model_dir:
+lora_r: 32
+lora_alpha: 16
+lora_dropout: 0.05
+lora_target_linear: true
+lora_fan_in_fan_out:
+
+gradient_accumulation_steps: 8
+micro_batch_size: 1
+eval_batch_size: 1
+num_epochs: 8
+optimizer: adamw_bnb_8bit
+lr_scheduler: cosine
+learning_rate: 1e-05
+
+max_grad_norm: 1.0
+logging_steps: 10
+
+train_on_inputs: false
+group_by_length: false
+
+bf16: auto
+fp16:
+tf32: false
+
+gradient_checkpointing: true
+gradient_checkpointing_kwargs:
+use_reentrant: false
+early_stopping_patience:
+resume_from_checkpoint:
+local_rank:
+xformers_attention:
+flash_attention: true
+warmup_ratio: 0.05
+eval_steps:
+save_strategy: epoch
+eval_table_size:
+num_processes: 8
+deepspeed:
+weight_decay: 0.0
+```
+
+</details><br>
+
+# home/ec2-user/SageMaker/task_decomposition/trained_models/granite-bfcl-plans-3.1-8b-lora
+
+This model is a fine-tuned version of [ibm-granite/granite-3.1-8b-instruct](https://huggingface.co/ibm-granite/granite-3.1-8b-instruct) on the bfcl_training_data.jsonl dataset.
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 1e-05
+- train_batch_size: 1
+- eval_batch_size: 1
+- seed: 42
+- distributed_type: multi-GPU
+- gradient_accumulation_steps: 8
+- total_train_batch_size: 64
+- total_eval_batch_size: 8
+- optimizer: Use adamw_bnb_8bit with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_steps: 20
+- num_epochs: 8
+
+### Training results
+
+
+
+### Framework versions
+
+- PEFT 0.13.2
+- Transformers 4.46.3
+- Pytorch 2.3.1+cu121
+- Datasets 3.1.0
+- Tokenizers 0.20.3

README.md

@@ -0,0 +1,140 @@
+---
+library_name: peft
+license: apache-2.0
+base_model: ibm-granite/granite-3.1-8b-instruct
+tags:
+- generated_from_trainer
+model-index:
+- name: granite-bfcl-plans-3.1-8b-lora
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+[<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
+<details><summary>See axolotl config</summary>
+
+axolotl version: `0.5.2`
+```yaml
+base_model: ibm-granite/granite-3.1-8b-instruct
+model_type: AutoModelForCausalLM
+tokenizer_type: AutoTokenizer
+
+resize_token_embeddings_to_32x: true
+load_in_8bit: true
+load_in_4bit: false
+strict: false
+
+datasets:
+- path: bfcl_training_data.jsonl
+  type: chat_template
+  chat_template: tokenizer_default
+  field_messages: conversations
+  message_field_role: role
+  message_field_content: value
+dataset_prepared_path: last_run_prepared_sft_bfcl
+
+val_set_size: 0
+sequence_len: 16384
+sample_packing: false
+pad_to_sequence_len: true
+eval_sample_packing: false
+output_dir: granite-bfcl-plans-3.1-8b-lora
+
+wandb_project: null
+wandb_entity: null
+wandb_watch: null
+wandb_name: null
+wandb_log_model: null
+
+adapter: lora
+lora_model_dir:
+lora_r: 32
+lora_alpha: 16
+lora_dropout: 0.05
+lora_target_linear: true
+lora_fan_in_fan_out:
+
+gradient_accumulation_steps: 8
+micro_batch_size: 1
+eval_batch_size: 1
+num_epochs: 8
+optimizer: adamw_bnb_8bit
+lr_scheduler: cosine
+learning_rate: 1e-05
+
+max_grad_norm: 1.0
+logging_steps: 10
+
+train_on_inputs: false
+group_by_length: false
+
+bf16: auto
+fp16:
+tf32: false
+
+gradient_checkpointing: true
+gradient_checkpointing_kwargs:
+use_reentrant: false
+early_stopping_patience:
+resume_from_checkpoint:
+local_rank:
+xformers_attention:
+flash_attention: true
+warmup_ratio: 0.05
+eval_steps:
+save_strategy: epoch
+eval_table_size:
+num_processes: 8
+deepspeed:
+weight_decay: 0.0
+```
+
+</details><br>
+
+# home/ec2-user/SageMaker/task_decomposition/trained_models/granite-bfcl-plans-3.1-8b-lora
+
+This model is a fine-tuned version of [ibm-granite/granite-3.1-8b-instruct](https://huggingface.co/ibm-granite/granite-3.1-8b-instruct) on the bfcl_training_data.jsonl dataset.
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 1e-05
+- train_batch_size: 1
+- eval_batch_size: 1
+- seed: 42
+- distributed_type: multi-GPU
+- gradient_accumulation_steps: 8
+- total_train_batch_size: 64
+- total_eval_batch_size: 8
+- optimizer: Use adamw_bnb_8bit with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_steps: 20
+- num_epochs: 8
+
+### Training results
+
+
+
+### Framework versions
+
+- PEFT 0.13.2
+- Transformers 4.46.3
+- Pytorch 2.3.1+cu121
+- Datasets 3.1.0
+- Tokenizers 0.20.3

adapter_config.json

@@ -0,0 +1,34 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": null,
+  "base_model_name_or_path": "ibm-granite/granite-3.1-8b-instruct",
+  "bias": "none",
+  "fan_in_fan_out": null,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layer_replication": null,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 16,
+  "lora_dropout": 0.05,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 32,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "gate_proj",
+    "v_proj",
+    "k_proj",
+    "o_proj",
+    "up_proj",
+    "q_proj",
+    "down_proj"
+  ],
+  "task_type": "CAUSAL_LM",
+  "use_dora": false,
+  "use_rslora": false
+}

adapter_model.bin

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

added_tokens.json

@@ -0,0 +1,5 @@
+{
+  "<|end_of_role|>": 49153,
+  "<|start_of_role|>": 49152,
+  "<|tool_call|>": 49154
+}

checkpoint-416/README.md

@@ -0,0 +1,202 @@
+---
+base_model: ibm-granite/granite-3.1-8b-instruct
+library_name: peft
+---
+
+# Model Card for Model ID
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+
+
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+
+
+- **Developed by:** [More Information Needed]
+- **Funded by [optional]:** [More Information Needed]
+- **Shared by [optional]:** [More Information Needed]
+- **Model type:** [More Information Needed]
+- **Language(s) (NLP):** [More Information Needed]
+- **License:** [More Information Needed]
+- **Finetuned from model [optional]:** [More Information Needed]
+
+### Model Sources [optional]
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** [More Information Needed]
+- **Paper [optional]:** [More Information Needed]
+- **Demo [optional]:** [More Information Needed]
+
+## Uses
+
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+### Direct Use
+
+<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+
+[More Information Needed]
+
+### Downstream Use [optional]
+
+<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
+
+[More Information Needed]
+
+### Out-of-Scope Use
+
+<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
+
+[More Information Needed]
+
+## Bias, Risks, and Limitations
+
+<!-- This section is meant to convey both technical and sociotechnical limitations. -->
+
+[More Information Needed]
+
+### Recommendations
+
+<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
+
+Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
+
+## How to Get Started with the Model
+
+Use the code below to get started with the model.
+
+[More Information Needed]
+
+## Training Details
+
+### Training Data
+
+<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
+
+[More Information Needed]
+
+### Training Procedure
+
+<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
+
+#### Preprocessing [optional]
+
+[More Information Needed]
+
+
+#### Training Hyperparameters
+
+- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
+
+#### Speeds, Sizes, Times [optional]
+
+<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
+
+[More Information Needed]
+
+## Evaluation
+
+<!-- This section describes the evaluation protocols and provides the results. -->
+
+### Testing Data, Factors & Metrics
+
+#### Testing Data
+
+<!-- This should link to a Dataset Card if possible. -->
+
+[More Information Needed]
+
+#### Factors
+
+<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
+
+[More Information Needed]
+
+#### Metrics
+
+<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+
+[More Information Needed]
+
+### Results
+
+[More Information Needed]
+
+#### Summary
+
+
+
+## Model Examination [optional]
+
+<!-- Relevant interpretability work for the model goes here -->
+
+[More Information Needed]
+
+## Environmental Impact
+
+<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
+
+Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
+
+- **Hardware Type:** [More Information Needed]
+- **Hours used:** [More Information Needed]
+- **Cloud Provider:** [More Information Needed]
+- **Compute Region:** [More Information Needed]
+- **Carbon Emitted:** [More Information Needed]
+
+## Technical Specifications [optional]
+
+### Model Architecture and Objective
+
+[More Information Needed]
+
+### Compute Infrastructure
+
+[More Information Needed]
+
+#### Hardware
+
+[More Information Needed]
+
+#### Software
+
+[More Information Needed]
+
+## Citation [optional]
+
+<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+
+**BibTeX:**
+
+[More Information Needed]
+
+**APA:**
+
+[More Information Needed]
+
+## Glossary [optional]
+
+<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+
+[More Information Needed]
+
+## More Information [optional]
+
+[More Information Needed]
+
+## Model Card Authors [optional]
+
+[More Information Needed]
+
+## Model Card Contact
+
+[More Information Needed]
+### Framework versions
+
+- PEFT 0.13.2

checkpoint-416/adapter_config.json

@@ -0,0 +1,34 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": null,
+  "base_model_name_or_path": "ibm-granite/granite-3.1-8b-instruct",
+  "bias": "none",
+  "fan_in_fan_out": null,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layer_replication": null,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 16,
+  "lora_dropout": 0.05,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 32,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "gate_proj",
+    "v_proj",
+    "k_proj",
+    "o_proj",
+    "up_proj",
+    "q_proj",
+    "down_proj"
+  ],
+  "task_type": "CAUSAL_LM",
+  "use_dora": false,
+  "use_rslora": false
+}

checkpoint-416/adapter_model.safetensors

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

checkpoint-416/added_tokens.json

@@ -0,0 +1,5 @@
+{
+  "<|end_of_role|>": 49153,
+  "<|start_of_role|>": 49152,
+  "<|tool_call|>": 49154
+}

checkpoint-416/optimizer.pt

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

checkpoint-416/rng_state_0.pth

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

checkpoint-416/rng_state_1.pth

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

checkpoint-416/rng_state_2.pth

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

checkpoint-416/rng_state_3.pth

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

checkpoint-416/rng_state_4.pth

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

checkpoint-416/rng_state_5.pth

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

checkpoint-416/rng_state_6.pth

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

checkpoint-416/rng_state_7.pth

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

checkpoint-416/scheduler.pt

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

checkpoint-416/special_tokens_map.json

@@ -0,0 +1,35 @@
+{
+  "additional_special_tokens": [
+    "<|start_of_role|>",
+    "<|end_of_role|>",
+    "<|tool_call|>"
+  ],
+  "bos_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-416/tokenizer_config.json

@@ -0,0 +1,199 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<|end_of_text|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<fim_prefix>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<fim_middle>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<fim_suffix>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "4": {
+      "content": "<fim_pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "5": {
+      "content": "<filename>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "6": {
+      "content": "<gh_stars>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "7": {
+      "content": "<issue_start>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "8": {
+      "content": "<issue_comment>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "9": {
+      "content": "<issue_closed>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "10": {
+      "content": "<jupyter_start>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "11": {
+      "content": "<jupyter_text>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "12": {
+      "content": "<jupyter_code>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "13": {
+      "content": "<jupyter_output>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "14": {
+      "content": "<empty_output>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "15": {
+      "content": "<commit_before>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "16": {
+      "content": "<commit_msg>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "17": {
+      "content": "<commit_after>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "18": {
+      "content": "<reponame>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49152": {
+      "content": "<|start_of_role|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49153": {
+      "content": "<|end_of_role|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49154": {
+      "content": "<|tool_call|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<|start_of_role|>",
+    "<|end_of_role|>",
+    "<|tool_call|>"
+  ],
+  "bos_token": "<|end_of_text|>",
+  "chat_template": "{%- if messages[0]['role'] == 'system' %}\n    {%- set system_message = messages[0]['content'] %}\n    {%- set loop_messages = messages[1:] %}\n{%- else %}\n    {%- set system_message = \"Knowledge Cutoff Date: April 2024.\nToday's Date: \" + strftime_now('%B %d, %Y') + \".\nYou are Granite, developed by IBM.\" %}\n    {%- if tools and documents %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant with access to the following tools. When a tool is required to answer the user's query, respond with <|tool_call|> followed by a JSON list of tools used. If a tool does not exist in the provided list of tools, notify the user that you do not have the ability to fulfill the request.\n\nWrite the response to the user's input by strictly aligning with the facts in the provided documents. If the information needed to answer the question is not available in the documents, inform the user that the question cannot be answered based on the available data.\" %}\n    {%- elif tools %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant with access to the following tools. When a tool is required to answer the user's query, respond with <|tool_call|> followed by a JSON list of tools used. If a tool does not exist in the provided list of tools, notify the user that you do not have the ability to fulfill the request.\" %}\n    {%- elif documents %}\n        {%- set system_message = system_message + \" Write the response to the user's input by strictly aligning with the facts in the provided documents. If the information needed to answer the question is not available in the documents, inform the user that the question cannot be answered based on the available data.\" %}\n    {%- else %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant.\" %}    \n    {%- endif %}\n    {%- if 'citations' in controls and documents %}\n        {%- set system_message = system_message + '\n\nIn your response, use the symbols <co> and </co> to indicate when a fact comes from a document in the search result, e.g <co>0</co> for a fact from document 0. Afterwards, list all the citations with their corresponding documents in an ordered list.' %}\n    {%- endif %}\n    {%- if 'hallucinations' in controls and documents %}\n        {%- set system_message = system_message + '\n\nFinally, after the response is written, include a numbered list of sentences from the response that are potentially hallucinated and not based in the documents.' %}\n    {%- endif %}\n    {%- set loop_messages = messages %}\n{%- endif %}\n{{- '<|start_of_role|>system<|end_of_role|>' + system_message + '<|end_of_text|>\n' }}\n{%- if tools %}\n    {{- '<|start_of_role|>tools<|end_of_role|>' }}\n    {{- tools | tojson(indent=4) }}\n    {{- '<|end_of_text|>\n' }}\n{%- endif %}\n{%- if documents %}\n    {{- '<|start_of_role|>documents<|end_of_role|>' }}\n    {%- for document in documents %}\n        {{- 'Document ' + loop.index0 | string + '\n' }}\n        {{- document['text'] }}\n        {%- if not loop.last %}\n            {{- '\n\n'}}\n        {%- endif%}\n    {%- endfor %}\n    {{- '<|end_of_text|>\n' }}\n{%- endif %}\n{%- for message in loop_messages %}\n    {{- '<|start_of_role|>' + message['role'] + '<|end_of_role|>' + message['content'] + '<|end_of_text|>\n' }}\n    {%- if loop.last and add_generation_prompt %}\n        {{- '<|start_of_role|>assistant' }}\n            {%- if controls %}\n                {{- ' ' + controls | tojson()}}\n            {%- endif %}\n        {{- '<|end_of_role|>' }}\n    {%- endif %}\n{%- endfor %}",
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<|end_of_text|>",
+  "errors": "replace",
+  "extra_special_tokens": {},
+  "model_max_length": 9223372036854775807,
+  "pad_token": "<|end_of_text|>",
+  "padding_side": "left",
+  "tokenizer_class": "GPT2Tokenizer",
+  "unk_token": "<|end_of_text|>",
+  "vocab_size": 49152
+}

checkpoint-416/trainer_state.json

@@ -0,0 +1,320 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 8.0,
+  "eval_steps": 500,
+  "global_step": 416,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.19230769230769232,
+      "grad_norm": 18.481584548950195,
+      "learning_rate": 5e-06,
+      "loss": 2.3883,
+      "step": 10
+    },
+    {
+      "epoch": 0.38461538461538464,
+      "grad_norm": 19.580347061157227,
+      "learning_rate": 1e-05,
+      "loss": 2.3213,
+      "step": 20
+    },
+    {
+      "epoch": 0.5769230769230769,
+      "grad_norm": 15.994805335998535,
+      "learning_rate": 9.984273879759713e-06,
+      "loss": 2.041,
+      "step": 30
+    },
+    {
+      "epoch": 0.7692307692307693,
+      "grad_norm": 7.965670108795166,
+      "learning_rate": 9.937194443381972e-06,
+      "loss": 1.5074,
+      "step": 40
+    },
+    {
+      "epoch": 0.9615384615384616,
+      "grad_norm": 7.664641380310059,
+      "learning_rate": 9.859057841617709e-06,
+      "loss": 1.2839,
+      "step": 50
+    },
+    {
+      "epoch": 1.1538461538461537,
+      "grad_norm": 4.538412094116211,
+      "learning_rate": 9.750355588704728e-06,
+      "loss": 1.1003,
+      "step": 60
+    },
+    {
+      "epoch": 1.3461538461538463,
+      "grad_norm": 3.505202293395996,
+      "learning_rate": 9.611771470522908e-06,
+      "loss": 0.9952,
+      "step": 70
+    },
+    {
+      "epoch": 1.5384615384615383,
+      "grad_norm": 3.503310441970825,
+      "learning_rate": 9.444177243274619e-06,
+      "loss": 0.9066,
+      "step": 80
+    },
+    {
+      "epoch": 1.7307692307692308,
+      "grad_norm": 8.761858940124512,
+      "learning_rate": 9.248627149747573e-06,
+      "loss": 0.8805,
+      "step": 90
+    },
+    {
+      "epoch": 1.9230769230769231,
+      "grad_norm": 3.437913417816162,
+      "learning_rate": 9.026351287655294e-06,
+      "loss": 0.8365,
+      "step": 100
+    },
+    {
+      "epoch": 2.1153846153846154,
+      "grad_norm": 4.105048656463623,
+      "learning_rate": 8.778747871771293e-06,
+      "loss": 0.8053,
+      "step": 110
+    },
+    {
+      "epoch": 2.3076923076923075,
+      "grad_norm": 3.81002140045166,
+      "learning_rate": 8.507374438531606e-06,
+      "loss": 0.7823,
+      "step": 120
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 3.294114351272583,
+      "learning_rate": 8.213938048432697e-06,
+      "loss": 0.7625,
+      "step": 130
+    },
+    {
+      "epoch": 2.6923076923076925,
+      "grad_norm": 6.447517395019531,
+      "learning_rate": 7.900284547855992e-06,
+      "loss": 0.7562,
+      "step": 140
+    },
+    {
+      "epoch": 2.8846153846153846,
+      "grad_norm": 9.523582458496094,
+      "learning_rate": 7.568386957867033e-06,
+      "loss": 0.7262,
+      "step": 150
+    },
+    {
+      "epoch": 3.076923076923077,
+      "grad_norm": 9.12429141998291,
+      "learning_rate": 7.2203330630288714e-06,
+      "loss": 0.7324,
+      "step": 160
+    },
+    {
+      "epoch": 3.269230769230769,
+      "grad_norm": 9.097236633300781,
+      "learning_rate": 6.858312278301638e-06,
+      "loss": 0.7087,
+      "step": 170
+    },
+    {
+      "epoch": 3.4615384615384617,
+      "grad_norm": 3.4165868759155273,
+      "learning_rate": 6.484601876641375e-06,
+      "loss": 0.6936,
+      "step": 180
+    },
+    {
+      "epoch": 3.6538461538461537,
+      "grad_norm": 7.2428741455078125,
+      "learning_rate": 6.101552663932704e-06,
+      "loss": 0.6814,
+      "step": 190
+    },
+    {
+      "epoch": 3.8461538461538463,
+      "grad_norm": 2.954061269760132,
+      "learning_rate": 5.711574191366427e-06,
+      "loss": 0.6859,
+      "step": 200
+    },
+    {
+      "epoch": 4.038461538461538,
+      "grad_norm": 3.99127197265625,
+      "learning_rate": 5.317119598282823e-06,
+      "loss": 0.6545,
+      "step": 210
+    },
+    {
+      "epoch": 4.230769230769231,
+      "grad_norm": 11.693526268005371,
+      "learning_rate": 4.9206701808259605e-06,
+      "loss": 0.6809,
+      "step": 220
+    },
+    {
+      "epoch": 4.423076923076923,
+      "grad_norm": 3.3279478549957275,
+      "learning_rate": 4.524719783479088e-06,
+      "loss": 0.6467,
+      "step": 230
+    },
+    {
+      "epoch": 4.615384615384615,
+      "grad_norm": 4.7100653648376465,
+      "learning_rate": 4.131759111665349e-06,
+      "loss": 0.6581,
+      "step": 240
+    },
+    {
+      "epoch": 4.8076923076923075,
+      "grad_norm": 3.9555509090423584,
+      "learning_rate": 3.7442600640946045e-06,
+      "loss": 0.6292,
+      "step": 250
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 3.8766026496887207,
+      "learning_rate": 3.3646601834128924e-06,
+      "loss": 0.6699,
+      "step": 260
+    },
+    {
+      "epoch": 5.1923076923076925,
+      "grad_norm": 3.9335460662841797,
+      "learning_rate": 2.995347322966933e-06,
+      "loss": 0.6512,
+      "step": 270
+    },
+    {
+      "epoch": 5.384615384615385,
+      "grad_norm": 3.7817418575286865,
+      "learning_rate": 2.6386446261365874e-06,
+      "loss": 0.6154,
+      "step": 280
+    },
+    {
+      "epoch": 5.576923076923077,
+      "grad_norm": 5.912929058074951,
+      "learning_rate": 2.296795912722014e-06,
+      "loss": 0.6366,
+      "step": 290
+    },
+    {
+      "epoch": 5.769230769230769,
+      "grad_norm": 3.369823694229126,
+      "learning_rate": 1.971951564311668e-06,
+      "loss": 0.6247,
+      "step": 300
+    },
+    {
+      "epoch": 5.961538461538462,
+      "grad_norm": 4.121553897857666,
+      "learning_rate": 1.6661549974185426e-06,
+      "loss": 0.6254,
+      "step": 310
+    },
+    {
+      "epoch": 6.153846153846154,
+      "grad_norm": 3.645986318588257,
+      "learning_rate": 1.3813298094746491e-06,
+      "loss": 0.6152,
+      "step": 320
+    },
+    {
+      "epoch": 6.346153846153846,
+      "grad_norm": 4.413705825805664,
+      "learning_rate": 1.1192676785412154e-06,
+      "loss": 0.6293,
+      "step": 330
+    },
+    {
+      "epoch": 6.538461538461538,
+      "grad_norm": 3.640406847000122,
+      "learning_rate": 8.816170928508367e-07,
+      "loss": 0.6167,
+      "step": 340
+    },
+    {
+      "epoch": 6.730769230769231,
+      "grad_norm": 3.553299903869629,
+      "learning_rate": 6.698729810778065e-07,
+      "loss": 0.6277,
+      "step": 350
+    },
+    {
+      "epoch": 6.923076923076923,
+      "grad_norm": 3.4969024658203125,
+      "learning_rate": 4.853673085668947e-07,
+      "loss": 0.6129,
+      "step": 360
+    },
+    {
+      "epoch": 7.115384615384615,
+      "grad_norm": 4.395870685577393,
+      "learning_rate": 3.2926069867446673e-07,
+      "loss": 0.6258,
+      "step": 370
+    },
+    {
+      "epoch": 7.3076923076923075,
+      "grad_norm": 4.367610931396484,
+      "learning_rate": 2.0253513192751374e-07,
+      "loss": 0.6197,
+      "step": 380
+    },
+    {
+      "epoch": 7.5,
+      "grad_norm": 4.083893775939941,
+      "learning_rate": 1.0598776892610685e-07,
+      "loss": 0.6052,
+      "step": 390
+    },
+    {
+      "epoch": 7.6923076923076925,
+      "grad_norm": 3.643474578857422,
+      "learning_rate": 4.02259358460233e-08,
+      "loss": 0.6206,
+      "step": 400
+    },
+    {
+      "epoch": 7.884615384615385,
+      "grad_norm": 3.4030489921569824,
+      "learning_rate": 5.6633040849601865e-09,
+      "loss": 0.6034,
+      "step": 410
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 416,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 8,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 2.1117165605182505e+19,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-416/training_args.bin

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

config.json

@@ -0,0 +1,49 @@
+{
+  "_attn_implementation_autoset": true,
+  "_name_or_path": "ibm-granite/granite-3.1-8b-instruct",
+  "architectures": [
+    "GraniteForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.1,
+  "attention_multiplier": 0.0078125,
+  "bos_token_id": 0,
+  "embedding_multiplier": 12.0,
+  "eos_token_id": 0,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 12800,
+  "logits_scaling": 16.0,
+  "max_position_embeddings": 131072,
+  "mlp_bias": false,
+  "model_type": "granite",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 40,
+  "num_key_value_heads": 8,
+  "pad_token_id": 0,
+  "quantization_config": {
+    "_load_in_4bit": false,
+    "_load_in_8bit": true,
+    "bnb_4bit_compute_dtype": "float32",
+    "bnb_4bit_quant_storage": "uint8",
+    "bnb_4bit_quant_type": "fp4",
+    "bnb_4bit_use_double_quant": false,
+    "llm_int8_enable_fp32_cpu_offload": false,
+    "llm_int8_has_fp16_weight": false,
+    "llm_int8_skip_modules": null,
+    "llm_int8_threshold": 6.0,
+    "load_in_4bit": false,
+    "load_in_8bit": true,
+    "quant_method": "bitsandbytes"
+  },
+  "residual_multiplier": 0.22,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000000.0,
+  "tie_word_embeddings": true,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.46.3",
+  "use_cache": false,
+  "vocab_size": 49184
+}

merged/added_tokens.json

@@ -0,0 +1,5 @@
+{
+  "<|end_of_role|>": 49153,
+  "<|start_of_role|>": 49152,
+  "<|tool_call|>": 49154
+}

merged/config.json

@@ -0,0 +1,33 @@
+{
+  "_name_or_path": "ibm-granite/granite-3.1-8b-instruct",
+  "architectures": [
+    "GraniteForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.1,
+  "attention_multiplier": 0.0078125,
+  "bos_token_id": 0,
+  "embedding_multiplier": 12.0,
+  "eos_token_id": 0,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 12800,
+  "logits_scaling": 16.0,
+  "max_position_embeddings": 131072,
+  "mlp_bias": false,
+  "model_type": "granite",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 40,
+  "num_key_value_heads": 8,
+  "pad_token_id": 0,
+  "residual_multiplier": 0.22,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000000.0,
+  "tie_word_embeddings": true,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.46.3",
+  "use_cache": false,
+  "vocab_size": 49184
+}

merged/generation_config.json

@@ -0,0 +1,8 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 0,
+  "do_sample": true,
+  "eos_token_id": 0,
+  "pad_token_id": 0,
+  "transformers_version": "4.46.3"
+}

merged/pytorch_model-00001-of-00004.bin

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

merged/pytorch_model-00002-of-00004.bin

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

merged/pytorch_model-00003-of-00004.bin

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

merged/pytorch_model-00004-of-00004.bin

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

merged/pytorch_model.bin.index.json

@@ -0,0 +1,370 @@
+{
+  "metadata": {
+    "total_size": 16341934080
+  },
+  "weight_map": {
+    "lm_head.weight": "pytorch_model-00001-of-00004.bin",
+    "model.embed_tokens.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.0.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.1.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.10.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.11.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.11.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.11.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.11.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.11.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.11.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.11.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.11.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.11.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.12.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.12.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.12.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.12.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.12.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.12.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.12.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.12.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.12.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.13.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.14.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.15.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.16.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.17.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.18.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.19.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.2.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.2.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.2.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.2.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.2.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.2.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.2.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.2.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.2.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.20.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.20.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.20.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.20.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.20.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.20.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.20.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.20.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.20.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.21.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.22.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.input_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.mlp.down_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.mlp.gate_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.mlp.up_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.post_attention_layernorm.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.self_attn.k_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.self_attn.o_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.self_attn.q_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.23.self_attn.v_proj.weight": "pytorch_model-00002-of-00004.bin",
+    "model.layers.24.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.24.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.24.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.24.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.24.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.24.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.24.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.24.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.24.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.25.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.26.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.27.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.28.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.29.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.3.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.3.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.3.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.3.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.3.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.3.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.3.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.3.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.3.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.30.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.30.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.30.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.30.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.30.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.30.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.30.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.30.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.30.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.31.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.32.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.33.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.34.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.input_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.mlp.down_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.mlp.up_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.post_attention_layernorm.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.35.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.36.input_layernorm.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.36.mlp.down_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.36.mlp.gate_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.36.mlp.up_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.36.post_attention_layernorm.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.36.self_attn.k_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.36.self_attn.o_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.36.self_attn.q_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.36.self_attn.v_proj.weight": "pytorch_model-00003-of-00004.bin",
+    "model.layers.37.input_layernorm.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.37.mlp.down_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.37.mlp.gate_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.37.mlp.up_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.37.post_attention_layernorm.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.37.self_attn.k_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.37.self_attn.o_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.37.self_attn.q_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.37.self_attn.v_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.input_layernorm.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.mlp.down_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.mlp.gate_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.mlp.up_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.post_attention_layernorm.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.self_attn.k_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.self_attn.o_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.self_attn.q_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.38.self_attn.v_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.input_layernorm.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.mlp.down_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.mlp.gate_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.mlp.up_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.post_attention_layernorm.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.self_attn.k_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.self_attn.o_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.self_attn.q_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.39.self_attn.v_proj.weight": "pytorch_model-00004-of-00004.bin",
+    "model.layers.4.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.4.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.4.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.4.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.4.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.4.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.4.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.4.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.4.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.5.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.6.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.7.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.8.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.input_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.mlp.down_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.mlp.gate_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.mlp.up_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.post_attention_layernorm.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.self_attn.k_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.self_attn.o_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.self_attn.q_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.layers.9.self_attn.v_proj.weight": "pytorch_model-00001-of-00004.bin",
+    "model.norm.weight": "pytorch_model-00004-of-00004.bin"
+  }
+}

merged/special_tokens_map.json

@@ -0,0 +1,35 @@
+{
+  "additional_special_tokens": [
+    "<|start_of_role|>",
+    "<|end_of_role|>",
+    "<|tool_call|>"
+  ],
+  "bos_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

merged/tokenizer_config.json

@@ -0,0 +1,199 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<|end_of_text|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<fim_prefix>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<fim_middle>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<fim_suffix>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "4": {
+      "content": "<fim_pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "5": {
+      "content": "<filename>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "6": {
+      "content": "<gh_stars>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "7": {
+      "content": "<issue_start>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "8": {
+      "content": "<issue_comment>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "9": {
+      "content": "<issue_closed>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "10": {
+      "content": "<jupyter_start>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "11": {
+      "content": "<jupyter_text>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "12": {
+      "content": "<jupyter_code>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "13": {
+      "content": "<jupyter_output>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "14": {
+      "content": "<empty_output>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "15": {
+      "content": "<commit_before>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "16": {
+      "content": "<commit_msg>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "17": {
+      "content": "<commit_after>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "18": {
+      "content": "<reponame>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49152": {
+      "content": "<|start_of_role|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49153": {
+      "content": "<|end_of_role|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49154": {
+      "content": "<|tool_call|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<|start_of_role|>",
+    "<|end_of_role|>",
+    "<|tool_call|>"
+  ],
+  "bos_token": "<|end_of_text|>",
+  "chat_template": "{%- if messages[0]['role'] == 'system' %}\n    {%- set system_message = messages[0]['content'] %}\n    {%- set loop_messages = messages[1:] %}\n{%- else %}\n    {%- set system_message = \"Knowledge Cutoff Date: April 2024.\nToday's Date: \" + strftime_now('%B %d, %Y') + \".\nYou are Granite, developed by IBM.\" %}\n    {%- if tools and documents %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant with access to the following tools. When a tool is required to answer the user's query, respond with <|tool_call|> followed by a JSON list of tools used. If a tool does not exist in the provided list of tools, notify the user that you do not have the ability to fulfill the request.\n\nWrite the response to the user's input by strictly aligning with the facts in the provided documents. If the information needed to answer the question is not available in the documents, inform the user that the question cannot be answered based on the available data.\" %}\n    {%- elif tools %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant with access to the following tools. When a tool is required to answer the user's query, respond with <|tool_call|> followed by a JSON list of tools used. If a tool does not exist in the provided list of tools, notify the user that you do not have the ability to fulfill the request.\" %}\n    {%- elif documents %}\n        {%- set system_message = system_message + \" Write the response to the user's input by strictly aligning with the facts in the provided documents. If the information needed to answer the question is not available in the documents, inform the user that the question cannot be answered based on the available data.\" %}\n    {%- else %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant.\" %}    \n    {%- endif %}\n    {%- if 'citations' in controls and documents %}\n        {%- set system_message = system_message + '\n\nIn your response, use the symbols <co> and </co> to indicate when a fact comes from a document in the search result, e.g <co>0</co> for a fact from document 0. Afterwards, list all the citations with their corresponding documents in an ordered list.' %}\n    {%- endif %}\n    {%- if 'hallucinations' in controls and documents %}\n        {%- set system_message = system_message + '\n\nFinally, after the response is written, include a numbered list of sentences from the response that are potentially hallucinated and not based in the documents.' %}\n    {%- endif %}\n    {%- set loop_messages = messages %}\n{%- endif %}\n{{- '<|start_of_role|>system<|end_of_role|>' + system_message + '<|end_of_text|>\n' }}\n{%- if tools %}\n    {{- '<|start_of_role|>tools<|end_of_role|>' }}\n    {{- tools | tojson(indent=4) }}\n    {{- '<|end_of_text|>\n' }}\n{%- endif %}\n{%- if documents %}\n    {{- '<|start_of_role|>documents<|end_of_role|>' }}\n    {%- for document in documents %}\n        {{- 'Document ' + loop.index0 | string + '\n' }}\n        {{- document['text'] }}\n        {%- if not loop.last %}\n            {{- '\n\n'}}\n        {%- endif%}\n    {%- endfor %}\n    {{- '<|end_of_text|>\n' }}\n{%- endif %}\n{%- for message in loop_messages %}\n    {{- '<|start_of_role|>' + message['role'] + '<|end_of_role|>' + message['content'] + '<|end_of_text|>\n' }}\n    {%- if loop.last and add_generation_prompt %}\n        {{- '<|start_of_role|>assistant' }}\n            {%- if controls %}\n                {{- ' ' + controls | tojson()}}\n            {%- endif %}\n        {{- '<|end_of_role|>' }}\n    {%- endif %}\n{%- endfor %}",
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<|end_of_text|>",
+  "errors": "replace",
+  "extra_special_tokens": {},
+  "model_max_length": 9223372036854775807,
+  "pad_token": "<|end_of_text|>",
+  "padding_side": "left",
+  "tokenizer_class": "GPT2Tokenizer",
+  "unk_token": "<|end_of_text|>",
+  "vocab_size": 49152
+}

special_tokens_map.json

@@ -0,0 +1,35 @@
+{
+  "additional_special_tokens": [
+    "<|start_of_role|>",
+    "<|end_of_role|>",
+    "<|tool_call|>"
+  ],
+  "bos_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<|end_of_text|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,199 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<|end_of_text|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<fim_prefix>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<fim_middle>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "3": {
+      "content": "<fim_suffix>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "4": {
+      "content": "<fim_pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "5": {
+      "content": "<filename>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "6": {
+      "content": "<gh_stars>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "7": {
+      "content": "<issue_start>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "8": {
+      "content": "<issue_comment>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "9": {
+      "content": "<issue_closed>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "10": {
+      "content": "<jupyter_start>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "11": {
+      "content": "<jupyter_text>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "12": {
+      "content": "<jupyter_code>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "13": {
+      "content": "<jupyter_output>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "14": {
+      "content": "<empty_output>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "15": {
+      "content": "<commit_before>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "16": {
+      "content": "<commit_msg>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "17": {
+      "content": "<commit_after>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "18": {
+      "content": "<reponame>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49152": {
+      "content": "<|start_of_role|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49153": {
+      "content": "<|end_of_role|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "49154": {
+      "content": "<|tool_call|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<|start_of_role|>",
+    "<|end_of_role|>",
+    "<|tool_call|>"
+  ],
+  "bos_token": "<|end_of_text|>",
+  "chat_template": "{%- if messages[0]['role'] == 'system' %}\n    {%- set system_message = messages[0]['content'] %}\n    {%- set loop_messages = messages[1:] %}\n{%- else %}\n    {%- set system_message = \"Knowledge Cutoff Date: April 2024.\nToday's Date: \" + strftime_now('%B %d, %Y') + \".\nYou are Granite, developed by IBM.\" %}\n    {%- if tools and documents %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant with access to the following tools. When a tool is required to answer the user's query, respond with <|tool_call|> followed by a JSON list of tools used. If a tool does not exist in the provided list of tools, notify the user that you do not have the ability to fulfill the request.\n\nWrite the response to the user's input by strictly aligning with the facts in the provided documents. If the information needed to answer the question is not available in the documents, inform the user that the question cannot be answered based on the available data.\" %}\n    {%- elif tools %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant with access to the following tools. When a tool is required to answer the user's query, respond with <|tool_call|> followed by a JSON list of tools used. If a tool does not exist in the provided list of tools, notify the user that you do not have the ability to fulfill the request.\" %}\n    {%- elif documents %}\n        {%- set system_message = system_message + \" Write the response to the user's input by strictly aligning with the facts in the provided documents. If the information needed to answer the question is not available in the documents, inform the user that the question cannot be answered based on the available data.\" %}\n    {%- else %}\n        {%- set system_message = system_message + \" You are a helpful AI assistant.\" %}    \n    {%- endif %}\n    {%- if 'citations' in controls and documents %}\n        {%- set system_message = system_message + '\n\nIn your response, use the symbols <co> and </co> to indicate when a fact comes from a document in the search result, e.g <co>0</co> for a fact from document 0. Afterwards, list all the citations with their corresponding documents in an ordered list.' %}\n    {%- endif %}\n    {%- if 'hallucinations' in controls and documents %}\n        {%- set system_message = system_message + '\n\nFinally, after the response is written, include a numbered list of sentences from the response that are potentially hallucinated and not based in the documents.' %}\n    {%- endif %}\n    {%- set loop_messages = messages %}\n{%- endif %}\n{{- '<|start_of_role|>system<|end_of_role|>' + system_message + '<|end_of_text|>\n' }}\n{%- if tools %}\n    {{- '<|start_of_role|>tools<|end_of_role|>' }}\n    {{- tools | tojson(indent=4) }}\n    {{- '<|end_of_text|>\n' }}\n{%- endif %}\n{%- if documents %}\n    {{- '<|start_of_role|>documents<|end_of_role|>' }}\n    {%- for document in documents %}\n        {{- 'Document ' + loop.index0 | string + '\n' }}\n        {{- document['text'] }}\n        {%- if not loop.last %}\n            {{- '\n\n'}}\n        {%- endif%}\n    {%- endfor %}\n    {{- '<|end_of_text|>\n' }}\n{%- endif %}\n{%- for message in loop_messages %}\n    {{- '<|start_of_role|>' + message['role'] + '<|end_of_role|>' + message['content'] + '<|end_of_text|>\n' }}\n    {%- if loop.last and add_generation_prompt %}\n        {{- '<|start_of_role|>assistant' }}\n            {%- if controls %}\n                {{- ' ' + controls | tojson()}}\n            {%- endif %}\n        {{- '<|end_of_role|>' }}\n    {%- endif %}\n{%- endfor %}",
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<|end_of_text|>",
+  "errors": "replace",
+  "extra_special_tokens": {},
+  "model_max_length": 9223372036854775807,
+  "pad_token": "<|end_of_text|>",
+  "padding_side": "left",
+  "tokenizer_class": "GPT2Tokenizer",
+  "unk_token": "<|end_of_text|>",
+  "vocab_size": 49152
+}